From 560bb4c29cbbfd940e2773df0ff28d0315b16517 Mon Sep 17 00:00:00 2001
From: Haidong Ji
Date: Sun, 9 Jun 2019 16:51:07 -0500
Subject: tree traversal done (in/pre/post) orders

Not too bad since I worked it out in Java first
---
 tests/tree_traversalTest.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 tests/tree_traversalTest.py

(limited to 'tests/tree_traversalTest.py')

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()
-- 
cgit v1.2.3