1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
/* Apache License, Version 2.0 */
#include "BLI_multi_value_map.hh"
#include "BLI_vector.hh"
#include "testing/testing.h"
namespace blender::tests {
TEST(multi_value_map, LookupNotExistant)
{
MultiValueMap<int, int> map;
EXPECT_EQ(map.lookup(5).size(), 0);
map.add(2, 5);
EXPECT_EQ(map.lookup(5).size(), 0);
}
TEST(multi_value_map, LookupExistant)
{
MultiValueMap<int, int> map;
map.add(2, 4);
map.add(2, 5);
map.add(3, 6);
EXPECT_EQ(map.lookup(2).size(), 2);
EXPECT_EQ(map.lookup(2)[0], 4);
EXPECT_EQ(map.lookup(2)[1], 5);
EXPECT_EQ(map.lookup(3).size(), 1);
EXPECT_EQ(map.lookup(3)[0], 6);
}
TEST(multi_value_map, AddMultiple)
{
MultiValueMap<int, int> map;
map.add_multiple(2, {4, 5, 6});
map.add_multiple(2, {1, 2});
map.add_multiple(5, {7, 5, 3});
EXPECT_EQ(map.lookup(2).size(), 5);
EXPECT_EQ(map.lookup(2)[0], 4);
EXPECT_EQ(map.lookup(2)[1], 5);
EXPECT_EQ(map.lookup(2)[2], 6);
EXPECT_EQ(map.lookup(2)[3], 1);
EXPECT_EQ(map.lookup(2)[4], 2);
EXPECT_EQ(map.lookup(5).size(), 3);
EXPECT_EQ(map.lookup(5)[0], 7);
EXPECT_EQ(map.lookup(5)[1], 5);
EXPECT_EQ(map.lookup(5)[2], 3);
}
TEST(multi_value_map, Keys)
{
MultiValueMap<int, int> map;
map.add(5, 7);
map.add(5, 7);
map.add_multiple(2, {6, 7, 8});
Vector<int> keys;
for (int key : map.keys()) {
keys.append(key);
}
EXPECT_EQ(keys.size(), 2);
EXPECT_TRUE(keys.contains(5));
EXPECT_TRUE(keys.contains(2));
}
TEST(multi_value_map, Values)
{
MultiValueMap<int, int> map;
map.add(3, 5);
map.add_multiple(3, {1, 2});
map.add(6, 1);
Vector<Span<int>> values;
for (Span<int> value_span : map.values()) {
values.append(value_span);
}
EXPECT_EQ(values.size(), 2);
}
TEST(multi_value_map, Items)
{
MultiValueMap<int, int> map;
map.add_multiple(4, {1, 2, 3});
for (auto &&item : map.items()) {
int key = item.key;
Span<int> values = item.value;
EXPECT_EQ(key, 4);
EXPECT_EQ(values.size(), 3);
EXPECT_EQ(values[0], 1);
EXPECT_EQ(values[1], 2);
EXPECT_EQ(values[2], 3);
}
}
TEST(multi_value_map, UniquePtr)
{
/* Mostly testing if it compiles here. */
MultiValueMap<std::unique_ptr<int>, std::unique_ptr<int>> map;
map.add(std::make_unique<int>(4), std::make_unique<int>(6));
map.add(std::make_unique<int>(4), std::make_unique<int>(7));
EXPECT_EQ(map.lookup(std::make_unique<int>(10)).size(), 0);
}
} // namespace blender::tests
|