class TestCTPMI_16_17(unittest.TestCase): def setUp(self): self.bst = BSTWithDFS() for n in [20, 10, 30, 5, 15, 3, 7, 17]: self.bst.insert(n) def test_case_01(self): node = self.bst.find_preorder(5) self.assertEqual(node.data, 5) self.assertEqual(node.left.data, 3) self.assertEqual(node.right.data, 7) def test_case_02(self): node = self.bst.find_inorder(30) self.assertEqual(node.data, 30) self.assertIsNone(node.left) self.assertIsNone(node.right) def test_case_03(self): node = self.bst.find_postorder(5) self.assertEqual(node.data, 5) self.assertEqual(node.left.data, 3) self.assertEqual(node.right.data, 7)
def setUp(self): self.bst = BSTWithDFS() for n in [20, 10, 30, 5, 15, 3, 7, 17]: self.bst.insert(n)