class TestBinarySearchTree(unittest.TestCase):
def __init__(self, *args, **kwargs):
#init method is inialized for each time that a test is run
super(TestBinarySearchTree, self).__init__(*args, **kwargs)
self.word = "hey"
self.binaryTree = BinaryTree(self.word)
self.words = [ "this", "is", "my", "exercise"]
for word in self.words:
self.binaryTree.insert_value(word)
def test_search(self):
x, path = self.binaryTree.find_word("this")
self.assertEqual(x, True)
x, path = self.binaryTree.find_word("not")
self.assertEqual(x, False)
def test_delete(self):
self.binaryTree.delete_node("is")
x, path = self.binaryTree.find_word("is")
self.assertFalse(x)
def test_pre_order(self):
print("Pre order Print")
self.binaryTree.print_pre_order()
