def test_bst_query_node_9_after_insert_10_and_9(self): bst = BinarySearchTree() bst.insert(10) bst.insert(9) answer = bst.query(9) self.assertEquals(answer.value, 9) self.assertEquals(answer, bst.root.left)
def test_bst_query_node_8_after_insert_7_and_8(self): bst = BinarySearchTree() bst.insert(7) bst.insert(8) answer = bst.query(8) self.assertEquals(answer.value, 8) self.assertEquals(answer, bst.root.right)
def test_bst_remove__single_tree_item(self): bst = BinarySearchTree() bst.insert(16) root = bst.remove(16) was_found = bst.query(16) self.assertEquals(root, None) self.assertEquals(was_found, False) self.assertEquals(bst.height, 0) self.assertEquals(bst.size, 0)
def test_bst_query_node_6_after_insert(self): bst = BinarySearchTree() bst.insert(6) answer = bst.query(6) self.assertEquals(answer, bst.root)
def test_bst_query_node_5_before_insert(self): bst = BinarySearchTree() answer = bst.query(5) self.assertEquals(answer, False)