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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
#define _GNU_SOURCE
#include "hashmap_tests.h"
#include "moslib/ds/hashmap.h"
#include <stdlib.h>
#include <string.h>
#include "moslib/ds/array.h"
#include "moslib/string.h"
Test put_get() {
HM_DEF(E, float, int);
E *hm = NULL;
hm_put(hm, 0.0f, 69);
hm_put(hm, 1.0f, 420);
test_assert(hm_get(hm, 0.0f) == 69, "hm_get got value not from hm_put");
test_assert(hm_get(hm, 1.0f) == 420, "hm_get got value not from hm_put");
hm_put(hm, 0.0f, 2137);
test_assert(hm_get(hm, 0.0f) == 2137, "hm_put didn't update value properly");
// a lot of puts and gets to test expansion of hashmap
for (int i = 2; i < 100; i++) {
hm_put(hm, (float)i, i);
}
int flag = 1;
for (int i = 2; i < 100; i++) {
if (hm_get(hm, (float)i) != i)
flag = 0;
}
test_assert(flag, "hm_put / hm_get failed when expansion was needed");
hm_free(hm);
}
Test put_e_get_e() {
HM_DEF(E, float, int);
E *hm = NULL;
hm_put_e(hm, (E){0.0f, 69});
hm_put_e(hm, (E){1.0f, 420});
E e = {0.0f, 69};
test_assert(memcmp(&e, hm_get_e(hm, 0.0f), sizeof(E)) == 0, "hm_get_e got entry not from hm_put_e");
e = (E){1.0f, 420};
test_assert(memcmp(&e, hm_get_e(hm, 1.0f), sizeof(E)) == 0, "hm_get_e got entry not from hm_put_e");
e = (E){1.0f, 2137};
hm_put_e(hm, e);
test_assert(memcmp(&e, hm_get_e(hm, 1.0f), sizeof(E)) == 0, "hm_put_e didn't update entry properly");
hm_free(hm);
}
Test str_key() {
HM_DEF(E, char *, int);
E *hm = hm_new(.key = STR);
char *str = "key";
char *str2 = strdup(str);
hm_put(hm, str, 69);
hm_put(hm, "hey", 420);
test_assert(hm_get(hm, str2) == 69, "hm_get got value not from hm_put");
test_assert(hm_get(hm, "hey") == 420, "hm_get got value not from hm_put");
hm_put(hm, str2, 2137);
test_assert(hm_get(hm, str) == 2137, "hm_put didn't update value properly");
free(str2);
hm_free(hm);
}
Test ss_key() {
HM_DEF(E, StringSlice, int);
E *hm = hm_new(.key = SS);
char *str = "key";
char *str2 = strdup(str);
StringSlice ss = ss_from_str(str);
StringSlice ss2 = ss_from_str(str2);
hm_put(hm, ss, 69);
hm_put(hm, SS_LIT("hey"), 420);
test_assert(hm_get(hm, ss2) == 69, "hm_get got value not from hm_put");
test_assert(hm_get(hm, SS_LIT("hey")) == 420, "hm_get got value not from hm_put");
hm_put(hm, ss2, 2137);
test_assert(hm_get(hm, ss) == 2137, "hm_put didn't update value properly");
free(str2);
hm_free(hm);
}
Test looping() {
HM_DEF(E, int, int);
E *hm = NULL;
hm_put(hm, 1, 10);
hm_put(hm, 2, 20);
hm_put(hm, 3, 30);
E *allowed = NULL;
E items[] = {{1, 10}, {2, 20}, {3, 30}};
arr_append_n(allowed, items, 3);
test_assert(memcmp(hm_first(hm), hm_next(hm, NULL), sizeof(E)) == 0, "hm_next(hm, NULL) != hm_first(hm)");
E *e = NULL;
while ((e = hm_next(hm, e))) {
int flag = 0;
for (size_t i = 0; i < arr_len(allowed); i++) {
if (memcmp(e, allowed + i, sizeof(E)) == 0) {
flag = 1;
arr_del(allowed, i);
break;
}
}
test_assert(flag, "hm_next produced an entry that wasn't put in the hashmap");
}
test_assert(arr_len(allowed) == 0, "hm_next didn't produce every entry from hashmap when it looped over it");
arr_free(allowed);
hm_free(hm);
}
void ds_hashmap_tests(Tester *tester) {
TestGroup *group = add_test_group(tester, "moslib/ds/hashmap.h");
add_test(group, put_get);
add_test(group, put_e_get_e);
add_test(group, str_key);
add_test(group, ss_key);
add_test(group, looping);
}
|
