def test_rbt_del_max():
rbt = RBT(None)
rbt.add(1)
rbt.add(3)
rbt.del_max()
assert rbt.find(3) == False
assert rbt.find(1) == True
def test_rbt_del_min():
rbt = RBT(None)
rbt.add(1)
assert rbt.find(1) == True
rbt.add(2)
assert rbt.find(2) == True
assert rbt.find(3) == False
# pudb.set_trace()
rbt.del_min()
assert rbt.find(1) == False
assert rbt.find(2) == True
def test_rbt():
rbt = RBT(None)
rbt.add(1)
assert rbt.root.val == 1
rbt.add(2)
assert rbt.root.val == 2
rbt.add(3)
assert rbt.root.val == 2
assert rbt.root.left.color == Color.BLACK
assert rbt.root.right.color == Color.BLACK
def test_rbt_delete():
rbt = RBT(None)
rbt.add(1)
rbt.add(2)
rbt.add(3)
rbt.delete(2)
assert rbt.find(2) == False
assert rbt.find(1) == True
assert rbt.find(3) == True
rbt.add(2)
rbt.delete(5)
assert rbt.find(1) == True
assert rbt.find(2) == True
assert rbt.find(3) == False