def test_delete_multiple_red_black(self, _, input_data, delete_data,
expected_result):
rb = RedBlackTree.build_from_list(input_data)
for key in delete_data:
rb.delete_by_key(key)
res = rb.iterative_inorder()
assert expected_result == res
def test_build_red_black(self, _, input_data, expected):
red_black = RedBlackTree.build_from_list(input_data)
inorder = red_black.iterative_inorder()
assert inorder == expected
def test_delete_in_red_black(self, _, input_data, key):
rb = RedBlackTree.build_from_list(input_data)
rb.delete_by_key(key)
res = set(rb.iterative_inorder())
assert key not in res
