@@ -0,0 +1,266 @@
|
||||
// vim:fileencoding=utf-8:foldmethod=marker
|
||||
|
||||
#include "wapp.h"
|
||||
#include "test_hashmap.h"
|
||||
|
||||
wp_persist inline WpU8Stream encoder(void *data);
|
||||
wp_persist inline b8 key_eq(void *a, void *b);
|
||||
|
||||
WpTestFuncResult test_hashmap(void) {
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 8);
|
||||
result = result && wpHashMapCapacity(&map) == 8 && wpHashMapCount(&map) == 0;
|
||||
map = wpHashMap(WpStr8, i32, encoder, key_eq, 3);
|
||||
result = result && wpHashMapCapacity(&map) == 4 && wpHashMapCount(&map) == 0;
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_alloc(void) {
|
||||
WpAllocator arena = wpMemArenaAllocatorInit(KiB(4));
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMapAlloc(WpStr8, i32, &arena, encoder, key_eq, 8);
|
||||
result = result && wpHashMapCapacity(&map) == 8 && wpHashMapCount(&map) == 0;
|
||||
map = wpHashMapAlloc(WpStr8, i32, &arena, encoder, key_eq, 3);
|
||||
result = result && wpHashMapCapacity(&map) == 4 && wpHashMapCount(&map) == 0;
|
||||
|
||||
wpMemArenaAllocatorDestroy(&arena);
|
||||
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_get_key_size(void) {
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 8);
|
||||
result = result && wpHashMapKeySize(&map) == sizeof(WpStr8);
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_get_value_size(void) {
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 8);
|
||||
result = result && wpHashMapValueSize(&map) == sizeof(i32);
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_find(void) {
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 8);
|
||||
WpStr8Array keys = wpArray(
|
||||
WpStr8,
|
||||
wpStr8Lit("key1"),
|
||||
wpStr8Lit("key2"),
|
||||
wpStr8Lit("key3"),
|
||||
wpStr8Lit("key4")
|
||||
);
|
||||
WpI32Array values = wpArray(i32, 1, 2, 3, 4);
|
||||
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
wpHashMapInsertCapped(WpStr8, i32, &map, &keys[i], &values[i]);
|
||||
}
|
||||
|
||||
WpHashLookupResult res = wpHashMapFind(WpStr8, &map, &keys[3]);
|
||||
|
||||
result = result && res.found && res.ref.dist_and_tag > 0;
|
||||
|
||||
WpStr8 key = wpStr8Lit("key0");
|
||||
res = wpHashMapFind(WpStr8, &map, &key);
|
||||
result = result && !res.found;
|
||||
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_update(void) {
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 4);
|
||||
WpStr8Array keys = wpArray(
|
||||
WpStr8,
|
||||
wpStr8Lit("key1"),
|
||||
wpStr8Lit("key2"),
|
||||
wpStr8Lit("key3"),
|
||||
wpStr8Lit("key4")
|
||||
);
|
||||
WpI32Array values = wpArray(i32, 1, 2, 3, 4);
|
||||
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
wpHashMapInsertCapped(WpStr8, i32, &map, &keys[i], &values[i]);
|
||||
}
|
||||
|
||||
i32 new_value = 0;
|
||||
wpHashMapUpdate(WpStr8, i32, &map, &keys[1], &new_value);
|
||||
result = result && wpHashMapCapacity(&map) == 4 && wpHashMapCount(&map) == 3 &&
|
||||
wpHashMapValues(i32, &map)[1] == new_value;
|
||||
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_insert_capped(void) {
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 4);
|
||||
WpStr8Array keys = wpArray(
|
||||
WpStr8,
|
||||
wpStr8Lit("key1"),
|
||||
wpStr8Lit("key2"),
|
||||
wpStr8Lit("key3"),
|
||||
wpStr8Lit("key4")
|
||||
);
|
||||
WpI32Array values = wpArray(i32, 1, 2, 3, 4);
|
||||
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
wpHashMapInsertCapped(WpStr8, i32, &map, &keys[i], &values[i]);
|
||||
}
|
||||
|
||||
result = result && wpHashMapCapacity(&map) == 4 && wpHashMapCount(&map) == 3;
|
||||
for (u64 i = 0; i < 3; ++i) {
|
||||
result = result && wpStr8Equal(wpArrayGet(WpStr8, map.keys, i), &keys[i]);
|
||||
}
|
||||
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_insert_or_update_capped(void) {
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 4);
|
||||
WpStr8Array keys = wpArray(
|
||||
WpStr8,
|
||||
wpStr8Lit("key1"),
|
||||
wpStr8Lit("key2"),
|
||||
wpStr8Lit("key3"),
|
||||
wpStr8Lit("key4")
|
||||
);
|
||||
WpI32Array values = wpArray(i32, 1, 2, 3, 4);
|
||||
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
wpHashMapInsertOrUpdateCapped(WpStr8, i32, &map, &keys[i], &values[i]);
|
||||
}
|
||||
|
||||
result = result && wpHashMapCapacity(&map) == 4 && wpHashMapCount(&map) == 3;
|
||||
for (u64 i = 0; i < 3; ++i) {
|
||||
result = result && wpStr8Equal(&wpHashMapKeys(WpStr8, &map)[i], &keys[i]);
|
||||
}
|
||||
|
||||
i32 new_value = 0;
|
||||
wpHashMapInsertOrUpdateCapped(WpStr8, i32, &map, &keys[1], &new_value);
|
||||
result = result && wpHashMapCapacity(&map) == 4 && wpHashMapCount(&map) == 3 &&
|
||||
wpHashMapValues(i32, &map)[1] == new_value;
|
||||
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_insert_alloc(void) {
|
||||
WpAllocator arena = wpMemArenaAllocatorInit(KiB(4));
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 4);
|
||||
WpStr8Array keys = wpArray(
|
||||
WpStr8,
|
||||
wpStr8Lit("key1"),
|
||||
wpStr8Lit("key2"),
|
||||
wpStr8Lit("key3"),
|
||||
wpStr8Lit("key4")
|
||||
);
|
||||
WpI32Array values = wpArray(i32, 1, 2, 3, 4);
|
||||
|
||||
WpStr8Array map_keys = map.keys;
|
||||
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
wpHashMapInsertAlloc(WpStr8, i32, &arena, &map, &keys[i], &values[i]);
|
||||
}
|
||||
|
||||
result = result && map.keys != map_keys && wpHashMapCapacity(&map) == 8 && wpHashMapCount(&map) == 4;
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
result = result && wpStr8Equal(&wpHashMapKeys(WpStr8, &map)[i], &keys[i]) &&
|
||||
wpHashMapValues(i32, &map)[i] == values[i];
|
||||
}
|
||||
|
||||
wpMemArenaAllocatorDestroy(&arena);
|
||||
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_insert_or_update_alloc(void) {
|
||||
WpAllocator arena = wpMemArenaAllocatorInit(KiB(4));
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 4);
|
||||
WpStr8Array keys = wpArray(
|
||||
WpStr8,
|
||||
wpStr8Lit("key1"),
|
||||
wpStr8Lit("key2"),
|
||||
wpStr8Lit("key3"),
|
||||
wpStr8Lit("key4")
|
||||
);
|
||||
WpI32Array values = wpArray(i32, 1, 2, 3, 4);
|
||||
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
wpHashMapInsertOrUpdateAlloc(WpStr8, i32, &arena, &map, &keys[i], &values[i]);
|
||||
}
|
||||
|
||||
result = result && wpHashMapCapacity(&map) == 8 && wpHashMapCount(&map) == 4;
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
result = result && wpStr8Equal(&wpHashMapKeys(WpStr8, &map)[i], &keys[i]);
|
||||
}
|
||||
|
||||
i32 new_value = 0;
|
||||
wpHashMapInsertOrUpdateAlloc(WpStr8, i32, &arena, &map, &keys[1], &new_value);
|
||||
result = result && wpHashMapCapacity(&map) == 8 && wpHashMapCount(&map) == 4 &&
|
||||
wpHashMapValues(i32, &map)[1] == new_value;
|
||||
|
||||
wpMemArenaAllocatorDestroy(&arena);
|
||||
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_remove(void) {
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 4);
|
||||
WpStr8Array keys = wpArray(
|
||||
WpStr8,
|
||||
wpStr8Lit("key1"),
|
||||
wpStr8Lit("key2"),
|
||||
wpStr8Lit("key3"),
|
||||
wpStr8Lit("key4")
|
||||
);
|
||||
WpI32Array values = wpArray(i32, 1, 2, 3, 4);
|
||||
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
wpHashMapInsertCapped(WpStr8, i32, &map, &keys[i], &values[i]);
|
||||
}
|
||||
|
||||
wpHashMapRemove(WpStr8, i32, &map, &keys[0]);
|
||||
|
||||
result = result && wpHashMapCapacity(&map) == 4 && wpHashMapCount(&map) == 2;
|
||||
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
WpTestFuncResult test_hashmap_clear(void) {
|
||||
b8 result = true;
|
||||
WpHashMap map = wpHashMap(WpStr8, i32, encoder, key_eq, 8);
|
||||
WpStr8Array keys = wpArray(
|
||||
WpStr8,
|
||||
wpStr8Lit("key1"),
|
||||
wpStr8Lit("key2"),
|
||||
wpStr8Lit("key3"),
|
||||
wpStr8Lit("key4")
|
||||
);
|
||||
WpI32Array values = wpArray(i32, 1, 2, 3, 4);
|
||||
|
||||
for (u64 i = 0; i < wpArrayCount(keys); ++i) {
|
||||
wpHashMapInsertCapped(WpStr8, i32, &map, &keys[i], &values[i]);
|
||||
}
|
||||
|
||||
wpHashMapClear(WpStr8, i32, &map);
|
||||
|
||||
result = result && wpHashMapCount(&map) == 0;
|
||||
|
||||
return wpTesterResult(result);
|
||||
}
|
||||
|
||||
wp_persist inline WpU8Stream encoder(void *data) {
|
||||
WpStr8 *str = (WpStr8 *)data;
|
||||
return wpStream(u8, str->buf, str->size);
|
||||
}
|
||||
|
||||
wp_persist inline b8 key_eq(void *a, void *b) {
|
||||
WpStr8 *a_str = (WpStr8 *)a;
|
||||
WpStr8 *b_str = (WpStr8 *)b;
|
||||
return wpStr8Equal(a_str, b_str);
|
||||
}
|
||||
Reference in New Issue
Block a user