def test_init_with_list_of_tuples(self): tree = TreeMap([(2, 'B'), (1, 'A'), (3, 'C')]) assert (tree.root.key, tree.root.value) == (2, 'B') assert (tree.root.left.key, tree.root.left.value) == (1, 'A') assert (tree.root.right.key, tree.root.right.value )== (3, 'C') assert tree.size == 3 assert tree.is_empty() is False
def test_delete(self): tree = TreeMap([(2, 'B'), (1, 'A'), (3, 'C')]) tree.delete(2) assert not tree.contains(2) assert tree.contains(1) assert tree.contains(3) tree.delete(1) assert not tree.contains(1) assert tree.contains(3) tree.delete(3) assert tree.is_empty() assert not tree.contains(3)
def test_init(self): tree = TreeMap() assert tree.root is None assert tree.size == 0 assert tree.is_empty() is True