def test_binary_tree_delete(self):
b = BinaryTree(key=5)
b.extend([2, 4, 6, 8, 10])
b.delete(6)
self.assertEquals(
'(5, None) -> ((2, None) -> (-, (4, None) -> (-, -)), (8, None) -> (-, (10, None) -> (-, -)))',
str(b)
)
b.delete(4)
self.assertEquals(
'(5, None) -> ((2, None) -> (-, -), (8, None) -> (-, (10, None) -> (-, -)))',
str(b)
)
b.delete(5)
self.assertEquals(
'(2, None) -> (-, -), (8, None) -> (-, (10, None) -> (-, -))',
str(b)
)
