def test_get():
m = TreeMap()
# permitsNullKeys=False, permitsNullValues=True, usesIdentity=False
put(m, 'A', True, None)
put(m, 'A', False, True)
put(m, 'B', True, None)
put(m, 'A', False, False)
try:
m.get(None)
pytest.fail('did not reject None key')
except TypeError:
pass
except Exception as e:
print('unexpected exception')
raise e
try:
m.put(None, True)
pytest.fail('did not reject None key')
except TypeError:
pass
except Exception as e:
print('unexpected exception')
raise e
try:
put(m, 'C', None, None)
put(m, 'C', True, None)
put(m, 'C', None, True)
except Exception as e:
print('unexpected exception')
raise e
def time(store):
tm = TreeMap()
for i in store:
tm.put(i, i)
start = time_ns()
for i in store:
tm.get(i)
end = time_ns()
return end - start
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)
def test():
m1 = TreeMap(none_at_end)
assert m1.put('a', 'a') is None
assert m1.put('b', 'b') is None
assert m1.put('c', 'c') is None
assert m1.put(None, 'd') is None
m2 = TreeMap(none_at_end)
m2.put_all(m1)
assert m1.last_key() is None
assert m1.get(m1.last_key()) == 'd'
assert m1.remove(m1.last_key()) == 'd'
assert m1.last_key() == 'c'
assert m2.entry_set().to_string() == '[a=a, b=b, c=c, None=d]'
m3 = m2.tail_map('b')
assert m3.last_key() is None
assert m3.get(m3.last_key()) == 'd'
assert m3.remove(m3.last_key()) == 'd'
assert m3.last_key() == 'c'
