def test():
comparable = TreeMap()
with pytest.raises(TypeError):
comparable.put(None, 'anything')
comparable.put('test', 'anything')
with pytest.raises(TypeError):
comparable.put(None, 'anything')
def case_insensitive_order(a, b):
x, y = a.lower(), b.lower()
if x < y:
return -1
elif x > y:
return 1
else:
return 0
comparator = TreeMap(case_insensitive_order)
with pytest.raises(Exception):
comparator.put(None, 'anything')
comparator.put('test', 'anything')
with pytest.raises(Exception):
comparator.put(None, 'anything')
comparator.clear()
with pytest.raises(Exception):
comparator.put(object(), 'anything')
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)
