Files
Containers/tst/test_unordered_map.c
Bailey Thompson e3bc87298f Simplify return values (#114)
Introduce bk_err and bk_bool rather than using int for both of these.
This means that the function definition now specifies whether 0 means
no error, or whether it means false.
2020-08-17 16:55:29 -04:00

356 lines
9.9 KiB
C

#include "test.h"
#include "../src/include/unordered_map.h"
static int compare_int(const void *const one, const void *const two)
{
const int a = *(int *) one;
const int b = *(int *) two;
return a - b;
}
static int hash_count;
static unsigned long hash_int(const void *const key)
{
unsigned long hash = 17;
hash_count++;
hash = 31 * hash + *(int *) key;
return hash;
}
static unsigned long bad_hash_int()
{
return 5;
}
static void test_invalid_init(void)
{
assert(!unordered_map_init(0, sizeof(int), hash_int, compare_int));
assert(!unordered_map_init(sizeof(int), 0, hash_int, compare_int));
assert(!unordered_map_init(sizeof(int), sizeof(int), NULL, compare_int));
assert(!unordered_map_init(sizeof(int), sizeof(int), hash_int, NULL));
}
static void test_put(unordered_map me)
{
int val_arr[10] = {5, 9, 4, -5, 0, 6, 1, 5, 7, 2};
int key;
int value;
int i;
int j;
assert(unordered_map_size(me) == 0);
assert(unordered_map_is_empty(me));
key = 4;
value = 9;
unordered_map_put(me, &key, &value);
assert(unordered_map_size(me) == 1);
value = 5;
unordered_map_put(me, &key, &value);
assert(unordered_map_size(me) == 1);
assert(!unordered_map_is_empty(me));
assert(unordered_map_contains(me, &key));
value = 0xfacade;
unordered_map_get(&value, me, &key);
assert(value == 5);
key = 7;
assert(!unordered_map_contains(me, &key));
unordered_map_put(me, &key, &value);
assert(unordered_map_size(me) == 2);
assert(unordered_map_contains(me, &key));
for (i = 0; i < 10; i++) {
unordered_map_put(me, &val_arr[i], &value);
assert(unordered_map_contains(me, &val_arr[i]));
}
assert(unordered_map_size(me) == 9);
for (i = 0; i < 10; i++) {
assert(unordered_map_contains(me, &val_arr[i]));
}
for (i = -100; i < 100; i++) {
int contains = 0;
for (j = 0; j < 10; j++) {
if (val_arr[j] == i) {
contains = 1;
}
}
assert(unordered_map_contains(me, &i) == contains);
}
}
static void test_remove(unordered_map me)
{
int num = -3;
assert(!unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 9);
assert(!unordered_map_contains(me, &num));
num = 6;
assert(unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 8);
assert(!unordered_map_contains(me, &num));
num = 4;
assert(unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 7);
assert(!unordered_map_contains(me, &num));
num = 7;
assert(unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 6);
assert(!unordered_map_contains(me, &num));
num = 9;
assert(unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 5);
assert(!unordered_map_contains(me, &num));
num = -5;
assert(unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 4);
assert(!unordered_map_contains(me, &num));
num = 0;
assert(unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 3);
assert(!unordered_map_contains(me, &num));
num = 1;
assert(unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 2);
assert(!unordered_map_contains(me, &num));
num = 5;
assert(unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 1);
assert(!unordered_map_contains(me, &num));
num = 2;
assert(unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 0);
assert(!unordered_map_contains(me, &num));
}
static void test_stress_remove(unordered_map me)
{
int value = 27;
int i;
for (i = 5000; i < 6000; i++) {
unordered_map_put(me, &i, &value);
assert(unordered_map_contains(me, &i));
}
assert(unordered_map_size(me) == 1000);
for (i = 5000; i < 6000; i++) {
unordered_map_remove(me, &i);
assert(!unordered_map_contains(me, &i));
}
assert(unordered_map_size(me) == 0);
assert(unordered_map_is_empty(me));
unordered_map_clear(me);
assert(unordered_map_size(me) == 0);
assert(unordered_map_is_empty(me));
}
static void test_stress_clear(unordered_map me)
{
int value = 57;
int key;
int i;
for (i = 5000; i < 6000; i++) {
unordered_map_put(me, &i, &value);
assert(unordered_map_contains(me, &i));
}
assert(unordered_map_size(me) == 1000);
hash_count = 0;
unordered_map_rehash(me);
assert(hash_count == 1000);
unordered_map_clear(me);
key = 0xfacade;
assert(!unordered_map_remove(me, &key));
assert(unordered_map_size(me) == 0);
assert(unordered_map_is_empty(me));
}
static void test_basic(void)
{
unordered_map me = unordered_map_init(sizeof(int), sizeof(int), hash_int,
compare_int);
assert(me);
test_put(me);
test_remove(me);
test_stress_remove(me);
test_stress_clear(me);
assert(!unordered_map_destroy(me));
}
static void test_bad_hash(void)
{
int num;
int key;
int value;
unordered_map me = unordered_map_init(sizeof(int), sizeof(int),
bad_hash_int, compare_int);
num = 1;
unordered_map_put(me, &num, &num);
num = 2;
unordered_map_put(me, &num, &num);
num = 3;
unordered_map_put(me, &num, &num);
num = 4;
unordered_map_put(me, &num, &num);
assert(unordered_map_size(me) == 4);
unordered_map_put(me, &num, &num);
assert(unordered_map_size(me) == 4);
assert(unordered_map_remove(me, &num));
assert(!unordered_map_remove(me, &num));
assert(unordered_map_size(me) == 3);
unordered_map_rehash(me);
assert(unordered_map_size(me) == 3);
key = 5;
value = 0xfacade;
assert(!unordered_map_get(&value, me, &key));
assert(value == 0xfacade);
assert(!unordered_map_destroy(me));
}
#if STUB_MALLOC
static void test_init_out_of_memory(void)
{
fail_malloc = 1;
assert(!unordered_map_init(sizeof(int), sizeof(int), bad_hash_int,
compare_int));
fail_calloc = 1;
assert(!unordered_map_init(sizeof(int), sizeof(int), bad_hash_int,
compare_int));
}
#endif
#if STUB_MALLOC
static void test_rehash_out_of_memory(void)
{
int key = 5;
int value = 7;
unordered_map me = unordered_map_init(sizeof(int), sizeof(int),
bad_hash_int, compare_int);
assert(me);
unordered_map_put(me, &key, &value);
assert(unordered_map_size(me) == 1);
assert(unordered_map_contains(me, &key));
fail_calloc = 1;
assert(unordered_map_rehash(me) == -ENOMEM);
assert(unordered_map_size(me) == 1);
assert(unordered_map_contains(me, &key));
assert(!unordered_map_destroy(me));
}
#endif
#if STUB_MALLOC
static void test_put_out_of_memory(void)
{
int key = 5;
int value = 7;
unordered_map me = unordered_map_init(sizeof(int), sizeof(int),
bad_hash_int, compare_int);
assert(me);
fail_malloc = 1;
assert(unordered_map_put(me, &key, &value) == -ENOMEM);
assert(unordered_map_put(me, &key, &value) == 0);
key = 7;
fail_malloc = 1;
assert(unordered_map_put(me, &key, &value) == -ENOMEM);
assert(!unordered_map_destroy(me));
}
#endif
#if STUB_MALLOC
static void test_resize_out_of_memory(void)
{
int i;
unordered_map me = unordered_map_init(sizeof(int), sizeof(int), hash_int,
compare_int);
for (i = 0; i < 11; i++) {
assert(unordered_map_put(me, &i, &i) == 0);
}
assert(unordered_map_size(me) == 11);
i++;
fail_calloc = 1;
assert(unordered_map_put(me, &i, &i) == -ENOMEM);
assert(unordered_map_size(me) == 11);
for (i = 0; i < 11; i++) {
assert(unordered_map_contains(me, &i));
}
assert(!unordered_map_destroy(me));
}
#endif
#if STUB_MALLOC
static void test_clear_out_of_memory(void)
{
int key = 5;
int value = 7;
unordered_map me = unordered_map_init(sizeof(int), sizeof(int), hash_int,
compare_int);
assert(me);
assert(unordered_map_put(me, &key, &value) == 0);
assert(unordered_map_size(me) == 1);
assert(unordered_map_contains(me, &key));
fail_calloc = 1;
assert(unordered_map_clear(me) == -ENOMEM);
assert(unordered_map_size(me) == 1);
assert(unordered_map_contains(me, &key));
assert(!unordered_map_destroy(me));
}
#endif
struct big_object {
int n;
double d;
signed char c[8];
};
static int compare_big_object(const void *const one, const void *const two)
{
const struct big_object *const a = one;
const struct big_object *const b = two;
return a->n - b->n;
}
static void test_big_object(void)
{
int i;
unordered_map me = unordered_map_init(sizeof(int),
sizeof(struct big_object),
bad_hash_int, compare_big_object);
assert(me);
for (i = 0; i < 16; i++) {
int j;
struct big_object b;
b.n = INT_MIN + i;
b.d = i + 0.5;
for (j = 0; j < 8; j++) {
b.c[j] = (signed char) (SCHAR_MIN + i + j);
}
assert(unordered_map_put(me, &i, &b) == 0);
b.n = -1;
b.d = -1;
for (j = 0; j < 8; j++) {
b.c[j] = -1;
}
}
for (i = 0; i < 16; i++) {
int j;
struct big_object b;
assert(unordered_map_get(&b, me, &i) == 1);
assert(b.n == INT_MIN + i);
assert(b.d == i + 0.5);
for (j = 0; j < 8; j++) {
assert(b.c[j] == SCHAR_MIN + i + j);
}
}
assert(!unordered_map_destroy(me));
}
void test_unordered_map(void)
{
test_invalid_init();
test_basic();
test_bad_hash();
#if STUB_MALLOC
test_init_out_of_memory();
test_rehash_out_of_memory();
test_put_out_of_memory();
test_resize_out_of_memory();
test_clear_out_of_memory();
#endif
test_big_object();
}