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