Example #1
0
    def test_add_right_child_success(self):
        tree = Tree(4, 50)
        tree.add_left(50, 25)
        tree.add_right(50, 75)

        result_data = [None] * 15
        result_data[0] = 50
        result_data[1] = 25
        result_data[2] = 75

        self.assertEqual(result_data, tree.data)
Example #2
0
    def setUp(self):
        tree = Tree(4, 50)
        tree.add_left(50, 25)
        tree.add_right(50, 75)

        tree.add_right(25, 37)
        tree.add_left(37, 31)
        tree.add_right(37, 43)

        tree.add_left(75, 62)
        tree.add_left(62, 55)

        tree.add_right(75, 84)
        tree.add_right(84, 92)

        self.tree = tree

        self.result_data = [
            50, 25, 75, None, 37, 62, 84, None, None, 31, 43, 55, None, None,
            92
        ]
Example #3
0
    def test_add_right_child_failed(self):
        tree = Tree(4, 50)

        with self.assertRaises(Exception):
            tree.add_right(99, 10)

        tree = Tree(1, 50)

        with self.assertRaises(Exception):
            tree.add_right(50, 10)

        tree = Tree(4, 50)

        with self.assertRaises(Exception):
            tree.add_right(50, 2)