From 457ab180d0f21a5ac547606065359d42d71cb173 Mon Sep 17 00:00:00 2001 From: Matej Focko Date: Sat, 7 May 2022 15:42:06 +0200 Subject: [PATCH] feat: add example usage of comparator Signed-off-by: Matej Focko --- comparator_usage.py | 47 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 47 insertions(+) create mode 100644 comparator_usage.py 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()