diff --git a/comparator_usage.py b/comparator_usage.py new file mode 100644 index 0000000..8f8c374 --- /dev/null +++ b/comparator_usage.py @@ -0,0 +1,47 @@ +from comparator import Comparator +from avl import AVLTree +from wavl import WAVLTree + +import logging +import random + + +def main(): + logging.basicConfig(level=logging.DEBUG) + + t = Comparator(AVLTree(), WAVLTree()) + # t = WAVLTree() + + values = [] + for _ in range(42): + x = random.randint(1, 100) + while x in values: + x = random.randint(1, 100) + + values.append(x) + t.insert(x) + + for _ in range(5): + x = random.choice(values) + values.remove(x) + t.delete(x) + + for _ in range(42): + x = random.randint(1, 100) + while x in values: + x = random.randint(1, 100) + + values.append(x) + t.insert(x) + + for _ in range(5): + x = random.choice(values) + values.remove(x) + t.delete(x) + + print(t) + # print(t.are_equal) + + +if __name__ == "__main__": + main()