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