def test():
m = TreeMap()
m.put(1, 1)
m.put(2, 2)
m.put(3, 3)
m2 = m.sub_map(2, 2)
with pytest.raises(KeyError):
m2.first_key()
with pytest.raises(KeyError):
m2.last_key()
m3 = m.sub_map(2, 3)
assert m3.first_key() == 2
assert m3.last_key() == 2
def test_main(self):
m = TreeMap()
self.test_navigable_map_removers(m)
self.test_navigable_map(m)
self.test_navigable_map(m.head_map(6, False))
self.test_navigable_map(m.head_map(5, True))
self.test_navigable_map(m.tail_map(0, False))
self.test_navigable_map(m.tail_map(1, True))
self.test_navigable_map(m.sub_map(1, 6, True, False))
self.test_navigable_map(m.sub_map(0, 5, False, True))
self.check_functional_invariants(m)
m.clear()
assert m.put(3333, 77777) is None
assert m.put(9134, 74982) is None
assert m.get(9134) == 74982
assert m.put(9134, 1382) == 74982
assert m.get(9134) == 1382
assert m.size() == 2
self.check_functional_invariants(m)