def __call__(self, fn) -> PASS_ARGS_T:
tree = get_ast(fn)
metadata = {}
if self.debug:
metadata["source_filename"] = inspect.getsourcefile(fn)
metadata["source_lines"] = inspect.getsourcelines(fn)
return tree, self.env, metadata
def test_get_ast():
def f():
pass
f_str = 'def f(): pass'
ast_str_0 = astor.dump_tree(get_ast(f))
ast_str_1 = astor.dump_tree(ast.parse(f_str).body[0])
assert ast_str_0 == ast_str_1