Test new map when used as a set.

map[K]struct{} works fine.
This commit is contained in:
Jeroen van Rijn
2022-11-15 01:27:29 +01:00
parent bbe44b49bc
commit 3949e2220f

View File

@@ -201,6 +201,120 @@ map_delete_random_key_value :: proc(t: ^testing.T) {
}
}
@test
set_insert_random_key_value :: proc(t: ^testing.T) {
seed_incr := u64(0)
for entries in ENTRY_COUNTS {
fmt.printf("[set_insert_random_key_value] Testing %v entries.\n", entries)
m: map[i64]struct{}
defer delete(m)
unique_keys := 0
r := rand.create(seed + seed_incr)
for _ in 0..<entries {
k := rand.int63(&r)
if k not_in m {
unique_keys += 1
}
m[k] = {}
}
key_count := 0
for k in m {
key_count += 1
}
expect(t, key_count == unique_keys, fmt.tprintf("Expected key_count to equal %v, got %v", unique_keys, key_count))
expect(t, len(m) == unique_keys, fmt.tprintf("Expected len(map) to equal %v, got %v", unique_keys, len(m)))
// Reset randomizer and verify
r = rand.create(seed + seed_incr)
num_fails := 0
for _ in 0..<entries {
k := rand.int63(&r)
cond := k in m
if !cond {
num_fails += 1
if num_fails > 5 {
fmt.println("... and more")
break
}
expect(t, false, fmt.tprintf("Unexpected value. Expected m[%v] to exist", k))
}
}
seed_incr += 1
}
}
@test
set_delete_random_key_value :: proc(t: ^testing.T) {
seed_incr := u64(0)
for entries in ENTRY_COUNTS {
fmt.printf("[set_delete_random_key_value] Testing %v entries.\n", entries)
m: map[i64]struct{}
defer delete(m)
unique_keys := 0
r := rand.create(seed + seed_incr)
for _ in 0..<entries {
k := rand.int63(&r)
if k not_in m {
unique_keys += 1
}
m[k] = {}
}
key_count := 0
for k in m {
key_count += 1
}
expect(t, key_count == unique_keys, fmt.tprintf("Expected key_count to equal %v, got %v", unique_keys, key_count))
expect(t, len(m) == unique_keys, fmt.tprintf("Expected len(map) to equal %v, got %v", unique_keys, len(m)))
half_entries := entries / 2
// Reset randomizer and delete half the entries
r = rand.create(seed + seed_incr)
for _ in 0..<half_entries {
k := rand.int63(&r)
delete_key(&m, k)
}
// Reset randomizer and verify
r = rand.create(seed + seed_incr)
num_fails := 0
for i in 0..<entries {
k := rand.int63(&r)
if i < half_entries {
if k in m {
num_fails += 1
if num_fails > 5 {
fmt.println("... and more")
break
}
expect(t, false, fmt.tprintf("Unexpected key present. Expected m[%v] to have been deleted", k))
}
} else {
if k not_in m {
num_fails += 1
if num_fails > 5 {
fmt.println("... and more")
break
}
expect(t, false, fmt.tprintf("Expected key not present. Expected m[%v] to exist", k))
}
}
}
seed_incr += 1
}
}
// -------- -------- -------- -------- -------- -------- -------- -------- -------- --------
main :: proc() {
@@ -219,6 +333,9 @@ main :: proc() {
mem_track_test(&t, map_update_random_key_value)
mem_track_test(&t, map_delete_random_key_value)
mem_track_test(&t, set_insert_random_key_value)
mem_track_test(&t, set_delete_random_key_value)
fmt.printf("%v/%v tests successful.\n", TEST_count - TEST_fail, TEST_count)
if TEST_fail > 0 {
os.exit(1)