def test_remove4():
tree = Tree()
s = 100
for i in reversed(range(0, s)):
tree.insert(i, i)
for i in reversed(range(0, s)):
tree.remove(i)
def test_remove3():
tree = Tree()
s = 100
for i in range(0, s):
tree.insert(i, i)
for i in range(0, s):
tree.remove(i)
def test_remove5():
tree = Tree()
l = list()
s = 100
for i in range(0, s):
tree.insert(i, i)
for i in range(0, s):
l.append(i)
random.shuffle(l)
size = 100
while len(l) > 1:
key = l.pop()
tree.remove(key)
size -= 1
assert len(tree.in_order()) == size
croot = tree.find(key)
assert croot.key != key
def test_remove2():
tree = Tree()
l = [38, 13, 51, 10, 25, 40, 84, 12, 37, 66, 89, 95]
length = len(l)
while len(l) > 0:
key = l.pop(0)
value = chr(key)
tree.insert(key, value)
tree.remove(100)
tree.remove(50)
tree.remove(14)
tree.remove(39)
assert len(tree.in_order()) == length
def test_remove():
tree = Tree()
l = [38, 13, 51, 10, 25, 40, 84, 12, 37, 66, 89, 95]
length = len(l)
while len(l) > 0:
key = l.pop(0)
value = chr(key)
tree.insert(key, value)
tree.remove(38)
assert tree.root.key == 37
assert tree.find(13).parent.key == 37
assert tree.find(51).parent.key == 37
assert tree.find(38).key == 40
tree.remove(89)
assert tree.find(95).key == 95
assert tree.find(95).parent.key == 84
tree.remove(13)
assert tree.find(13).key == 25
assert tree.find(12).parent.key == 37
assert tree.find(10).parent.key == 12
assert tree.find(25).parent.key == 12
