Function process_source_code

tests/test_ast_rewrite.py:10–18

(src: str, single=True)

def process_source_code(src: str, single=True) -> NodeType:
    tree = collect(dedent(src), minimal=True)
    loc = ScanLocation(location="<unknown>")

    v = Visitor.run_stages(location=loc, stages=("convert", "rewrite"), ast_tree=tree)
    if single:
        return v.tree[-1]
    else:
        return v.tree


def test_var_propagation_call_argument():