def test_count_root(self):
root = BSTNode(10, 10, None)
tree = BST(root)
node = BSTNode(15, 10, None)
tree.AddKeyValue(15, 10)
self.assertEqual(tree.Count(), 2)
def test_delete_root(self):
root = BSTNode(10, 10, None)
tree = BST(root)
tree.DeleteNodeByKey(10)
self.assertEqual(tree.Count(), 0)
