Python process_file Examples

Example #1

File: test_parenthesized_return.py Project: pavelpy/pybetter

def test_removal_of_parentheses_in_return(original, expected):
    processed, _ = process_file(original.strip(), [FixParenthesesInReturn])

    assert processed.strip() == (expected or original).strip()

Example #2

def test_collapse_of_nested_with_statements(original, expected):
    processed, _ = process_file(original.strip(), [FixTrivialNestedWiths])

    assert processed.strip() == (expected or original).strip()

Example #3

File: test_all_attribute.py Project: lensvol/pybetter

def test_generation_of_dunder_all(original, expected):
    processed, _ = process_file(original.strip(), [FixMissingAllAttribute])

    assert processed.strip() == (expected or original).strip()

Example #4

def test_trivial_fmt_string_conversion(original, expected):
    processed, _ = process_file(original.strip(),
                                [FixTrivialFmtStringCreation])

    assert processed.strip() == (expected or original).strip()

Example #5

File: test_boolean_equality.py Project: pavelpy/pybetter

def test_boolean_equality_fix(original, expected):
    processed, _ = process_file(original, [FixBooleanEqualityChecks])

    assert processed == expected

Example #6

File: test_not_in.py Project: pavelpy/pybetter

def test_not_in_transformation(original, expected):
    processed, _ = process_file(original.strip(), [FixNotInConditionOrder])

    assert processed.strip() == (expected or original).strip()

Example #7

File: test_equals_none.py Project: pavelpy/pybetter

def test_trivial_fmt_string_conversion(original, expected):
    processed, _ = process_file(original.strip(), [FixEqualsNone])

    assert processed.strip() == (expected or original).strip()

Example #8

def test_replacement_of_list_literal_in_sets(original, expected):
    processed, _ = process_file(original.strip(), [FixUnhashableList])

    assert processed.strip() == (expected or original).strip()

Example #9

File: test_no_crashes.py Project: lensvol/pybetter

def test_no_crashes_on_valid_input(generated_source):
    process_file(generated_source, ALL_IMPROVEMENTS)

Example #10

def test_mutable_defaults_extraction(original, expected):
    processed, _ = process_file(original.strip(), [FixMutableDefaultArgs])
    assert processed.strip() == (expected or original).strip()