def test_stack_overflow(self):
""" Inserts 10000 consecutive nodes which should cause stack overflow in the recursive implementation.
"""
sorted_binary_tree = SortedBinaryTree()
for key in range(5000, 10000):
self.assertFalse(sorted_binary_tree.contains(key))
sorted_binary_tree.add(key=key, payload=str(key))
for key in range(5000):
self.assertFalse(sorted_binary_tree.contains(key))
sorted_binary_tree.add(key=key, payload=str(key))
for expected_key, node in zip(range(10000),
sorted_binary_tree.enumerate()):
self.assertTrue(expected_key == node.key)
for key in range(5000, 10000):
sorted_binary_tree.get(key)
sorted_binary_tree.pop(key)
self.assertTrue(True)
def test_sanity(self):
""" This test will exercise all functions of the tree by: inserting, looking up, removing keys.
We will retain auxillary (and less efficient) data structures to ensure correctness.
"""
INSERT_NEW = 0
INSERT_EXISTING = 1
LOOKUP_EXISTING = 2
LOOKUP_NONEXISTENT = 4
POP_EXISTING = 8
POP_NONEXISTENT = 16
OPERATIONS = [
INSERT_NEW,
INSERT_NEW,
INSERT_NEW,
INSERT_NEW,
INSERT_NEW,
INSERT_EXISTING,
LOOKUP_EXISTING,
LOOKUP_NONEXISTENT,
POP_EXISTING,
POP_NONEXISTENT,
]
random.seed(2)
num_iterations = 1000
key_range = 100 * num_iterations
sorted_binary_tree = SortedBinaryTree()
inserted = set()
removed = set()
for i in range(num_iterations):
# Each case will need one of the below
existing_key = random.choice(list(inserted)) if inserted else None
new_key = random.randint(0, key_range)
while new_key in inserted:
new_key = random.randint(0, key_range)
OPERATION_TYPE = random.choice(OPERATIONS)
while existing_key is None and OPERATION_TYPE in (INSERT_EXISTING,
LOOKUP_EXISTING,
POP_EXISTING):
OPERATION_TYPE = random.choice(OPERATIONS)
if OPERATION_TYPE == INSERT_NEW:
sorted_binary_tree.add(new_key, str(new_key))
inserted.add(new_key)
if new_key in removed:
# If we re-inserted a new key, let's make sure to account for it.
removed.remove(new_key)
elif OPERATION_TYPE == INSERT_EXISTING:
try:
sorted_binary_tree.add(existing_key)
self.assertTrue(False)
except:
self.assertTrue(True)
elif OPERATION_TYPE == LOOKUP_EXISTING:
self.assertTrue(sorted_binary_tree.contains(existing_key))
self.assertTrue(
sorted_binary_tree.get(existing_key) == str(existing_key))
continue # No need to check invariants for read-only operations
elif OPERATION_TYPE == LOOKUP_NONEXISTENT:
self.assertFalse(sorted_binary_tree.contains(new_key))
try:
sorted_binary_tree.get(new_key)
self.assertTrue(False)
except:
self.assertTrue(True)
elif OPERATION_TYPE == POP_EXISTING:
self.assertTrue(sorted_binary_tree.contains(existing_key))
self.assertTrue(
sorted_binary_tree.pop(existing_key) == str(existing_key))
self.assertFalse(sorted_binary_tree.contains(existing_key))
inserted.remove(existing_key)
removed.add(existing_key)
elif OPERATION_TYPE == POP_NONEXISTENT:
self.assertFalse(sorted_binary_tree.contains(new_key))
try:
sorted_binary_tree.get(new_key)
self.assertTrue(False)
except:
self.assertTrue(True)
else:
assert False
# let's validate invariants
## let's make sure that all keys are present
for key in inserted:
self.assertTrue(sorted_binary_tree.contains(key))
## let's make sure that all removed keys are not present
for key in removed:
self.assertFalse(sorted_binary_tree.contains(key))
## let's make sure that the ordering is maintained
if not sorted_binary_tree.is_empty():
for set_key, tree_node in zip(sorted(inserted),
sorted_binary_tree.enumerate()):
self.assertTrue(set_key == tree_node.key)