|
@@ -3,14 +3,16 @@ package test_core_container
|
|
|
import rb "core:container/rbtree"
|
|
|
import "core:math/rand"
|
|
|
import "core:testing"
|
|
|
-import "core:fmt"
|
|
|
import "base:intrinsics"
|
|
|
import "core:mem"
|
|
|
import "core:slice"
|
|
|
-import tc "tests:common"
|
|
|
+import "core:log"
|
|
|
|
|
|
-RANDOM_SEED :: #config(RANDOM_SEED, 0)
|
|
|
-random_seed := u64(intrinsics.read_cycle_counter()) when RANDOM_SEED == 0 else u64(RANDOM_SEED)
|
|
|
+@(private)
|
|
|
+_RANDOM_SEED :: #config(RANDOM_SEED, u64(0))
|
|
|
+
|
|
|
+// Exported
|
|
|
+random_seed := u64(intrinsics.read_cycle_counter()) when _RANDOM_SEED == 0 else u64(_RANDOM_SEED)
|
|
|
|
|
|
test_rbtree_integer :: proc(t: ^testing.T, $Key: typeid, $Value: typeid) {
|
|
|
track: mem.Tracking_Allocator
|
|
@@ -21,15 +23,15 @@ test_rbtree_integer :: proc(t: ^testing.T, $Key: typeid, $Value: typeid) {
|
|
|
r: rand.Rand
|
|
|
rand.init(&r, random_seed)
|
|
|
|
|
|
- tc.log(t, fmt.tprintf("Testing Red-Black Tree($Key=%v,$Value=%v), using random seed %v, add -define:RANDOM_SEED=%v to reuse it.", type_info_of(Key), type_info_of(Value), random_seed, random_seed))
|
|
|
+ log.infof("Testing Red-Black Tree($Key=%v,$Value=%v), using random seed %v, add -define:RANDOM_SEED=%v to reuse it.", type_info_of(Key), type_info_of(Value), random_seed, random_seed)
|
|
|
tree: rb.Tree(Key, Value)
|
|
|
rb.init(&tree)
|
|
|
|
|
|
- tc.expect(t, rb.len(&tree) == 0, "empty: len should be 0")
|
|
|
- tc.expect(t, rb.first(&tree) == nil, "empty: first should be nil")
|
|
|
- tc.expect(t, rb.last(&tree) == nil, "empty: last should be nil")
|
|
|
+ testing.expect(t, rb.len(&tree) == 0, "empty: len should be 0")
|
|
|
+ testing.expect(t, rb.first(&tree) == nil, "empty: first should be nil")
|
|
|
+ testing.expect(t, rb.last(&tree) == nil, "empty: last should be nil")
|
|
|
iter := rb.iterator(&tree, .Forward)
|
|
|
- tc.expect(t, rb.iterator_get(&iter) == nil, "empty/iterator: first node should be nil")
|
|
|
+ testing.expect(t, rb.iterator_get(&iter) == nil, "empty/iterator: first node should be nil")
|
|
|
|
|
|
// Test insertion.
|
|
|
NR_INSERTS :: 32 + 1 // Ensure at least 1 collision.
|
|
@@ -45,27 +47,27 @@ test_rbtree_integer :: proc(t: ^testing.T, $Key: typeid, $Value: typeid) {
|
|
|
|
|
|
existing_node, in_map := inserted_map[k]
|
|
|
n, inserted, _ := rb.find_or_insert(&tree, k, v)
|
|
|
- tc.expect(t, in_map != inserted, "insert: inserted should match inverse of map lookup")
|
|
|
+ testing.expect(t, in_map != inserted, "insert: inserted should match inverse of map lookup")
|
|
|
if inserted {
|
|
|
inserted_map[k] = n
|
|
|
} else {
|
|
|
- tc.expect(t, existing_node == n, "insert: expecting existing node")
|
|
|
+ testing.expect(t, existing_node == n, "insert: expecting existing node")
|
|
|
}
|
|
|
}
|
|
|
|
|
|
entry_count := len(inserted_map)
|
|
|
- tc.expect(t, rb.len(&tree) == entry_count, "insert: len after")
|
|
|
+ testing.expect(t, rb.len(&tree) == entry_count, "insert: len after")
|
|
|
validate_rbtree(t, &tree)
|
|
|
|
|
|
first := rb.first(&tree)
|
|
|
last := rb.last(&tree)
|
|
|
- tc.expect(t, first != nil && first.key == min_key, fmt.tprintf("insert: first should be present with key %v", min_key))
|
|
|
- tc.expect(t, last != nil && last.key == max_key, fmt.tprintf("insert: last should be present with key %v", max_key))
|
|
|
+ testing.expectf(t, first != nil && first.key == min_key, "insert: first should be present with key %v", min_key)
|
|
|
+ testing.expectf(t, last != nil && last.key == max_key, "insert: last should be present with key %v", max_key)
|
|
|
|
|
|
// Ensure that all entries can be found.
|
|
|
for k, v in inserted_map {
|
|
|
- tc.expect(t, v == rb.find(&tree, k), "Find(): Node")
|
|
|
- tc.expect(t, k == v.key, "Find(): Node key")
|
|
|
+ testing.expect(t, v == rb.find(&tree, k), "Find(): Node")
|
|
|
+ testing.expect(t, k == v.key, "Find(): Node key")
|
|
|
}
|
|
|
|
|
|
// Test the forward/backward iterators.
|
|
@@ -79,21 +81,21 @@ test_rbtree_integer :: proc(t: ^testing.T, $Key: typeid, $Value: typeid) {
|
|
|
visited: int
|
|
|
for node in rb.iterator_next(&iter) {
|
|
|
k, idx := node.key, visited
|
|
|
- tc.expect(t, inserted_keys[idx] == k, "iterator/forward: key")
|
|
|
- tc.expect(t, node == rb.iterator_get(&iter), "iterator/forward: get")
|
|
|
+ testing.expect(t, inserted_keys[idx] == k, "iterator/forward: key")
|
|
|
+ testing.expect(t, node == rb.iterator_get(&iter), "iterator/forward: get")
|
|
|
visited += 1
|
|
|
}
|
|
|
- tc.expect(t, visited == entry_count, "iterator/forward: visited")
|
|
|
+ testing.expect(t, visited == entry_count, "iterator/forward: visited")
|
|
|
|
|
|
slice.reverse(inserted_keys[:])
|
|
|
iter = rb.iterator(&tree, rb.Direction.Backward)
|
|
|
visited = 0
|
|
|
for node in rb.iterator_next(&iter) {
|
|
|
k, idx := node.key, visited
|
|
|
- tc.expect(t, inserted_keys[idx] == k, "iterator/backward: key")
|
|
|
+ testing.expect(t, inserted_keys[idx] == k, "iterator/backward: key")
|
|
|
visited += 1
|
|
|
}
|
|
|
- tc.expect(t, visited == entry_count, "iterator/backward: visited")
|
|
|
+ testing.expect(t, visited == entry_count, "iterator/backward: visited")
|
|
|
|
|
|
// Test removal (and on_remove callback)
|
|
|
rand.shuffle(inserted_keys[:], &r)
|
|
@@ -104,19 +106,19 @@ test_rbtree_integer :: proc(t: ^testing.T, $Key: typeid, $Value: typeid) {
|
|
|
}
|
|
|
for k, i in inserted_keys {
|
|
|
node := rb.find(&tree, k)
|
|
|
- tc.expect(t, node != nil, "remove: find (pre)")
|
|
|
+ testing.expect(t, node != nil, "remove: find (pre)")
|
|
|
|
|
|
ok := rb.remove(&tree, k)
|
|
|
- tc.expect(t, ok, "remove: succeeds")
|
|
|
- tc.expect(t, entry_count - (i + 1) == rb.len(&tree), "remove: len (post)")
|
|
|
+ testing.expect(t, ok, "remove: succeeds")
|
|
|
+ testing.expect(t, entry_count - (i + 1) == rb.len(&tree), "remove: len (post)")
|
|
|
validate_rbtree(t, &tree)
|
|
|
|
|
|
- tc.expect(t, nil == rb.find(&tree, k), "remove: find (post")
|
|
|
+ testing.expect(t, nil == rb.find(&tree, k), "remove: find (post")
|
|
|
}
|
|
|
- tc.expect(t, rb.len(&tree) == 0, "remove: len should be 0")
|
|
|
- tc.expect(t, callback_count == 0, fmt.tprintf("remove: on_remove should've been called %v times, it was %v", entry_count, callback_count))
|
|
|
- tc.expect(t, rb.first(&tree) == nil, "remove: first should be nil")
|
|
|
- tc.expect(t, rb.last(&tree) == nil, "remove: last should be nil")
|
|
|
+ testing.expect(t, rb.len(&tree) == 0, "remove: len should be 0")
|
|
|
+ testing.expectf(t, callback_count == 0, "remove: on_remove should've been called %v times, it was %v", entry_count, callback_count)
|
|
|
+ testing.expect(t, rb.first(&tree) == nil, "remove: first should be nil")
|
|
|
+ testing.expect(t, rb.last(&tree) == nil, "remove: last should be nil")
|
|
|
|
|
|
// Refill the tree.
|
|
|
for k in inserted_keys {
|
|
@@ -130,32 +132,32 @@ test_rbtree_integer :: proc(t: ^testing.T, $Key: typeid, $Value: typeid) {
|
|
|
k := node.key
|
|
|
|
|
|
ok := rb.iterator_remove(&iter)
|
|
|
- tc.expect(t, ok, "iterator/remove: success")
|
|
|
+ testing.expect(t, ok, "iterator/remove: success")
|
|
|
|
|
|
ok = rb.iterator_remove(&iter)
|
|
|
- tc.expect(t, !ok, "iterator/remove: redundant removes should fail")
|
|
|
+ testing.expect(t, !ok, "iterator/remove: redundant removes should fail")
|
|
|
|
|
|
- tc.expect(t, rb.find(&tree, k) == nil, "iterator/remove: node should be gone")
|
|
|
- tc.expect(t, rb.iterator_get(&iter) == nil, "iterator/remove: get should return nil")
|
|
|
+ testing.expect(t, rb.find(&tree, k) == nil, "iterator/remove: node should be gone")
|
|
|
+ testing.expect(t, rb.iterator_get(&iter) == nil, "iterator/remove: get should return nil")
|
|
|
|
|
|
// Ensure that iterator_next still works.
|
|
|
node, ok = rb.iterator_next(&iter)
|
|
|
- tc.expect(t, ok == (rb.len(&tree) > 0), "iterator/remove: next should return false")
|
|
|
- tc.expect(t, node == rb.first(&tree), "iterator/remove: next should return first")
|
|
|
+ testing.expect(t, ok == (rb.len(&tree) > 0), "iterator/remove: next should return false")
|
|
|
+ testing.expect(t, node == rb.first(&tree), "iterator/remove: next should return first")
|
|
|
|
|
|
validate_rbtree(t, &tree)
|
|
|
}
|
|
|
- tc.expect(t, rb.len(&tree) == entry_count - 1, "iterator/remove: len should drop by 1")
|
|
|
+ testing.expect(t, rb.len(&tree) == entry_count - 1, "iterator/remove: len should drop by 1")
|
|
|
|
|
|
rb.destroy(&tree)
|
|
|
- tc.expect(t, rb.len(&tree) == 0, "destroy: len should be 0")
|
|
|
- tc.expect(t, callback_count == 0, fmt.tprintf("remove: on_remove should've been called %v times, it was %v", entry_count, callback_count))
|
|
|
+ testing.expect(t, rb.len(&tree) == 0, "destroy: len should be 0")
|
|
|
+ testing.expectf(t, callback_count == 0, "remove: on_remove should've been called %v times, it was %v", entry_count, callback_count)
|
|
|
|
|
|
// print_tree_node(tree._root)
|
|
|
delete(inserted_map)
|
|
|
delete(inserted_keys)
|
|
|
- tc.expect(t, len(track.allocation_map) == 0, fmt.tprintf("Expected 0 leaks, have %v", len(track.allocation_map)))
|
|
|
- tc.expect(t, len(track.bad_free_array) == 0, fmt.tprintf("Expected 0 bad frees, have %v", len(track.bad_free_array)))
|
|
|
+ testing.expectf(t, len(track.allocation_map) == 0, "Expected 0 leaks, have %v", len(track.allocation_map))
|
|
|
+ testing.expectf(t, len(track.bad_free_array) == 0, "Expected 0 bad frees, have %v", len(track.bad_free_array))
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -194,7 +196,7 @@ validate_rbtree :: proc(t: ^testing.T, tree: ^$T/rb.Tree($Key, $Value)) {
|
|
|
}
|
|
|
|
|
|
verify_rbtree_propery_1 :: proc(t: ^testing.T, n: ^$N/rb.Node($Key, $Value)) {
|
|
|
- tc.expect(t, rb.node_color(n) == .Black || rb.node_color(n) == .Red, "Property #1: Each node is either red or black.")
|
|
|
+ testing.expect(t, rb.node_color(n) == .Black || rb.node_color(n) == .Red, "Property #1: Each node is either red or black.")
|
|
|
if n == nil {
|
|
|
return
|
|
|
}
|
|
@@ -203,14 +205,14 @@ verify_rbtree_propery_1 :: proc(t: ^testing.T, n: ^$N/rb.Node($Key, $Value)) {
|
|
|
}
|
|
|
|
|
|
verify_rbtree_propery_2 :: proc(t: ^testing.T, root: ^$N/rb.Node($Key, $Value)) {
|
|
|
- tc.expect(t, rb.node_color(root) == .Black, "Property #2: Root node should be black.")
|
|
|
+ testing.expect(t, rb.node_color(root) == .Black, "Property #2: Root node should be black.")
|
|
|
}
|
|
|
|
|
|
verify_rbtree_propery_4 :: proc(t: ^testing.T, n: ^$N/rb.Node($Key, $Value)) {
|
|
|
if rb.node_color(n) == .Red {
|
|
|
// A red node's left, right and parent should be black
|
|
|
all_black := rb.node_color(n._left) == .Black && rb.node_color(n._right) == .Black && rb.node_color(n._parent) == .Black
|
|
|
- tc.expect(t, all_black, "Property #3: Red node's children + parent must be black.")
|
|
|
+ testing.expect(t, all_black, "Property #3: Red node's children + parent must be black.")
|
|
|
}
|
|
|
if n == nil {
|
|
|
return
|
|
@@ -233,7 +235,7 @@ verify_rbtree_propery_5_helper :: proc(t: ^testing.T, n: ^$N/rb.Node($Key, $Valu
|
|
|
if path_black_count^ == -1 {
|
|
|
path_black_count^ = black_count
|
|
|
} else {
|
|
|
- tc.expect(t, black_count == path_black_count^, "Property #5: Paths from a node to its leaves contain same black count.")
|
|
|
+ testing.expect(t, black_count == path_black_count^, "Property #5: Paths from a node to its leaves contain same black count.")
|
|
|
}
|
|
|
return
|
|
|
}
|