1 files changed, 37 insertions, 0 deletions
diff --git a/tests/tree_traversalTest.py b/tests/tree_traversalTest.py
new file mode 100644
index 0000000..7fe79e3
--- /dev/null
+++ b/tests/tree_traversalTest.py
@@ -0,0 +1,37 @@
+import unittest
+
+from sources.tree_traversal import in_order, pre_order, post_order
+
+class MyTestCase(unittest.TestCase):
+    def test_something(self):
+        key = [4, 2, 5, 1, 3]
+        left = [1, 3, -1, -1, -1]
+        right = [2, 4, -1, -1, -1]
+
+        result = in_order(key, left, right)
+        self.assertEqual(len(result), 5)
+        self.assertEqual(result[0], 1)
+        self.assertEqual(result[1], 2)
+        self.assertEqual(result[2], 3)
+        self.assertEqual(result[3], 4)
+        self.assertEqual(result[4], 5)
+
+        result = pre_order(key, left, right)
+        self.assertEqual(len(result), 5)
+        self.assertEqual(result[0], 4)
+        self.assertEqual(result[1], 2)
+        self.assertEqual(result[2], 1)
+        self.assertEqual(result[3], 3)
+        self.assertEqual(result[4], 5)
+
+        result = post_order(key, left, right)
+        self.assertEqual(len(result), 5)
+        self.assertEqual(result[0], 1)
+        self.assertEqual(result[1], 3)
+        self.assertEqual(result[2], 2)
+        self.assertEqual(result[3], 5)
+        self.assertEqual(result[4], 4)
+
+
+if __name__ == '__main__':
+    unittest.main()