def test_variables_finder(): tree = ast.parse(''' let(a) x = 1 let(b) ''') assert find_variables(tree) == ['a', 'b']
def my_snippet(class_name, x_value): class class_name: pass let(x) x = x_value let(result) result = 0 let(i) for i in range(x): result += i return result