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_delete(self):
ht = TreeMap()
ht.set('I', 1)
ht.set('V', 5)
ht.set('X', 10)
assert ht.length() == 3
assert ht.size == 3
ht.delete('I')
ht.delete('X')
assert ht.length() == 1
assert ht.size == 1
with self.assertRaises(KeyError):
ht.delete('X') # Key no longer exists
with self.assertRaises(KeyError):
ht.delete('A') # Key does not exist
def test_delete(self):
tm = TreeMap()
assert tm.size == 0
tm.set(1, 2)
assert tm.size == 1
tm.set(2, 3)
assert tm.size == 2
tm.set(3, 4)
assert tm.size == 3
self.assertCountEqual(tm.items(), [(1, 2), (2, 3), (3, 4)])
tm.delete(3)
assert tm.size == 2
self.assertCountEqual(tm.items(), [(1, 2), (2, 3)])
tm.delete(2)
assert tm.size == 1
tm.delete(1)
assert tm.size == 0
