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