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