def test_remove_zero(self):
size = 100
input_arr = [0] * size
for i in range(size):
input_arr[i] = randint(-10, 10)
tree = SplayTree()
for v in input_arr:
tree.add(v)
arr = []
for i in range(size):
if input_arr[i] != 0:
arr.append(input_arr[i])
arr.sort()
tree.remove(0)
j = 0
checked = True
for p in tree:
if p != arr[j]:
checked = False
break
j += 1
self.assertTrue(checked)
def test_contains1(self):
size = 1000
input_arr = [0] * size
for i in range(size):
input_arr[i] = randint(-1000, 1000)
tree = SplayTree()
for v in input_arr:
tree.add(v)
value = 1001
self.assertFalse(tree.contains(value))
def _check(self, input_arr):
tree = SplayTree()
for v in input_arr:
tree.add(v)
input_arr.sort()
i = 0
for p in tree:
if p != input_arr[i]:
return False
i += 1
return True
