diff --git a/Programming_Paradigms/bintree_test.py b/Programming_Paradigms/bintree_test.py index 38ce41a8c88b6db37412c709bd5752fe4003ad8b..998ad90aa69d2be1221e61dbebd42f24939c85a6 100644 --- a/Programming_Paradigms/bintree_test.py +++ b/Programming_Paradigms/bintree_test.py @@ -12,8 +12,8 @@ class TestBinTree(unittest.TestCase): def test_has_insert_and_search(self): tree = BinTree() tree.insert(1) - self.assertTrue(tree.search(1)) - self.assertFalse(tree.search(0)) + self.assertIsInstance(tree.search(1), BinTreeNode) + self.assertIsNone(tree.search(0)) def test_insert_into_empty_tree(self): tree = BinTree()