def test_removal_of_parentheses_in_return(original, expected): processed, _ = process_file(original.strip(), [FixParenthesesInReturn]) assert processed.strip() == (expected or original).strip()
def test_collapse_of_nested_with_statements(original, expected): processed, _ = process_file(original.strip(), [FixTrivialNestedWiths]) assert processed.strip() == (expected or original).strip()
def test_generation_of_dunder_all(original, expected): processed, _ = process_file(original.strip(), [FixMissingAllAttribute]) assert processed.strip() == (expected or original).strip()
def test_trivial_fmt_string_conversion(original, expected): processed, _ = process_file(original.strip(), [FixTrivialFmtStringCreation]) assert processed.strip() == (expected or original).strip()
def test_boolean_equality_fix(original, expected): processed, _ = process_file(original, [FixBooleanEqualityChecks]) assert processed == expected
def test_not_in_transformation(original, expected): processed, _ = process_file(original.strip(), [FixNotInConditionOrder]) assert processed.strip() == (expected or original).strip()
def test_trivial_fmt_string_conversion(original, expected): processed, _ = process_file(original.strip(), [FixEqualsNone]) assert processed.strip() == (expected or original).strip()
def test_replacement_of_list_literal_in_sets(original, expected): processed, _ = process_file(original.strip(), [FixUnhashableList]) assert processed.strip() == (expected or original).strip()
def test_no_crashes_on_valid_input(generated_source): process_file(generated_source, ALL_IMPROVEMENTS)
def test_mutable_defaults_extraction(original, expected): processed, _ = process_file(original.strip(), [FixMutableDefaultArgs]) assert processed.strip() == (expected or original).strip()