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