def test_delete_node_one_child(self):
"""Test to delete a node that has only one child."""
#Tree should look like this
# 23
# 12 34
# 9 15 45
# 8 56
bst = BinarySearchTree()
self.assertEqual(bst.size, 0)
self.assertEqual(bst.get_height(), -1)
nodes = [23, 12, 34, 9, 15, 45, 56, 8]
#add nodes
for node in nodes:
bst.insert(node)
self.assertEqual(bst.size, 8)
self.assertEqual(bst.get_height(), 3)
#delete node that just has a left child
bst.delete(9)
self.assertEqual(bst.size, 7)
self.assertEqual(bst.get_height(), 3)
self.assertIsNone(bst.search(9))
#delete node that just has a right child
bst.delete(45)
self.assertEqual(bst.size, 6)
self.assertEqual(bst.get_height(), 2)
self.assertIsNone(bst.search(45))
def test_search(self):
"""Test that the search method returns the correct
nodes.
"""
bst = BinarySearchTree()
bst.insert(12)
bst.insert(9)
bst.insert(22)
#look for an element that doesn't exist
node = bst.search(200)
self.assertIsNone(node)
#look for an element that does exist
node = bst.search(12)
self.assertIsNotNone(node)
self.assertEqual(node.data, 12)
#look for the root's left child
node = bst.search(9)
self.assertIsNotNone(node)
self.assertEqual(node.data, 9)
#look for the root's right child
node = bst.search(22)
self.assertIsNotNone(node)
self.assertEqual(node.data, 22)
def test_delete_root_node_two_children(self):
"""Test to ensure the delete method correctly
deletes a root node that has two children.
"""
#Tree should look like this
# 23
# 12 34
# 9 15 45
# 8 56
bst = BinarySearchTree()
self.assertEqual(bst.size, 0)
self.assertEqual(bst.get_height(), -1)
nodes = [23, 12, 34, 9, 15, 45, 56, 8]
#add nodes
for node in nodes:
bst.insert(node)
self.assertEqual(bst.size, 8)
self.assertEqual(bst.get_height(), 3)
bst.delete(23)
self.assertEqual(bst.size, 7)
self.assertEqual(bst.get_height(), 3)
self.assertIsNone(bst.search(23))
self.assertEqual(bst.root.data, 34)
def test_delete_root_node_left_child(self):
"""Test to delete a root node with only one child
that is the left child.
"""
#Tree should look like this
# 56
# 45
# 34
# 23
bst = BinarySearchTree()
self.assertEqual(bst.size, 0)
self.assertEqual(bst.get_height(), -1)
nodes = [56, 45, 34, 23]
#add nodes
for node in nodes:
bst.insert(node)
self.assertEqual(bst.size, 4)
self.assertEqual(bst.get_height(), 3)
#root has only a right child
bst.delete(56)
self.assertEqual(bst.size, 3)
self.assertEqual(bst.get_height(), 2)
self.assertIsNone(bst.search(56))
self.assertEqual(bst.root.data, 45)
def test_delete_leaf_node(self):
"""Test that the delete method functions correctly
when deleting a leaf node.
"""
#Tree should look like this
# 23
# 12 34
# 9 15 45
# 56
bst = BinarySearchTree()
self.assertEqual(bst.size, 0)
self.assertEqual(bst.get_height(), -1)
nodes = [23, 12, 34, 9, 15, 45, 56]
#add nodes
for node in nodes:
bst.insert(node)
self.assertEqual(bst.size, 7)
self.assertEqual(bst.get_height(), 3)
#Case 1: Deleting a leaf node
bst.delete(56)
self.assertEqual(bst.size, 6)
self.assertEqual(bst.get_height(), 2)
self.assertIsNone(bst.search(7))
def test_delete_root_as_leaf(self):
"""Test to ensure that the delete method works properly
when the root is the only node in the tree.
"""
bst = BinarySearchTree()
self.assertEqual(bst.size, 0)
self.assertEqual(bst.get_height(), -1)
bst.insert(12)
self.assertEqual(bst.size, 1)
self.assertEqual(bst.get_height(), 0)
root = bst.search(12)
self.assertEqual(root.data, 12)
bst.delete(12)
self.assertEqual(bst.size, 0)
self.assertEqual(bst.get_height(), -1)
self.assertIsNone(bst.root)
def test_delete_node_two_children(self):
"""Test to delete a node with two children."""
#Tree should look like this
# 23
# 12 34
# 9 15 45
# 8 56
bst = BinarySearchTree()
self.assertEqual(bst.size, 0)
self.assertEqual(bst.get_height(), -1)
nodes = [23, 12, 34, 9, 15, 45, 56, 8]
#add nodes
for node in nodes:
bst.insert(node)
self.assertEqual(bst.size, 8)
self.assertEqual(bst.get_height(), 3)
bst.delete(12)
self.assertEqual(bst.size, 7)
self.assertEqual(bst.get_height(), 3)
self.assertIsNone(bst.search(12))