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()