from RedBlackTree import *
# test for randomly insertion and deletion
treeRBT = RedBlackTree()
num_nodes = 20
# insertion
key_list = [random.randint(1000, 2000) for i in range(num_nodes)]
value_list = [random.randint(0, 100) for i in range(1, num_nodes + 1)]
random.shuffle(key_list)
print("key list: {}".format(key_list))
print("value list: {}".format(value_list))
for i in range(num_nodes):
treeRBT.insert(key_list[i], value_list[i])
treeRBT.check_all()
treeRBT.show_paths()
# show all information of all nodes
# for i in treeRBT:
# i.show_info()
# deletion
delete_list = [i for i in range(num_nodes)]
random.shuffle(delete_list)
for i in range(num_nodes):
delete_index = delete_list[i]
treeRBT.delete(key_list[delete_index])
print("delete {} : {}".format(key_list[delete_index], value_list[delete_index]))
treeRBT.check_all(output_information=False)
