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)
