diff --git a/Programming_Paradigms/bintree_test.py b/Programming_Paradigms/bintree_test.py index 998ad90aa69d2be1221e61dbebd42f24939c85a6..1141370b9ac881ed449d48b395b948bb47c637ce 100644 --- a/Programming_Paradigms/bintree_test.py +++ b/Programming_Paradigms/bintree_test.py @@ -22,6 +22,20 @@ class TestBinTree(unittest.TestCase): self.assertIsInstance(tree.root, BinTreeNode) self.assertEqual(tree.root.value, 1) + def test_example(self): + values = [7, 3, 9, 2, 4, 8, 11] + tree = BinTree() + for value in values: + tree.insert(value) + for value in values: + node = tree.search(value) + self.assertIsInstance(node, BinTreeNode) + self.assertEqual(value, node.value) + node = tree.search(3) + self.assertEqual(node.value, 3) + self.assertEqual(node.left.value, 2) + self.assertEqual(node.right.value, 4) + class TestBinTreeNode(unittest.TestCase): def test_creation(self): node = BinTreeNode(5)