def test_items(self):
ht = TreeMap()
assert ht.items() == []
ht.set('I', 1)
assert ht.items() == [('I', 1)]
ht.set('V', 5)
self.assertCountEqual(ht.items(), [('I', 1), ('V', 5)])
ht.set('X', 10)
self.assertCountEqual(ht.items(), [('I', 1), ('V', 5), ('X', 10)])
def test_set(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.set(3, 5)
assert tm.size == 3
self.assertCountEqual(tm.items(), [(1, 2), (2, 3), (3, 5)])
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
def test_items(self):
tm = TreeMap({"c": "C", "b": "B", "a": "A"})
keys = [k for k, v in tm.items()]
self.assertSequenceEqual(["a", "b", "c"], keys)
def test_items(self):
tree = TreeMap([(2, 'B'), (1, 'A'), (3, 'C')])
assert tree.items() == [(1, 'A'), (2, 'B'), (3, 'C')]
def test_init_with_list(self):
tm = TreeMap([(1, 2), (2, 3), (3, 4)])
assert tm.size == 3
self.assertCountEqual(tm.items(), [(1, 2), (2, 3), (3, 4)])
