def test_parse_file(self): path = parse.__file__ stem, _ = os.path.splitext(path) path = '%s.py' % stem with open(path) as stream: source = stream.read() parsed = parse.parse(source, path) self.assertTrue(isinstance(parsed, ast.AST))
def re_render(module_text): tree = parse(module_text) recast_docstrings(tree) adjust_lines(tree) return render(tree)
def _parse(self): from pym.ast.parse import parse return parse(self)
def test_parse_string(self): source = 'i = 0\ni += 1' parsed = parse.parse(source) self.assertTrue(isinstance(parsed, ast.AST))
def test_render(self): expected = 'i = 0\ni += 1' node = parse(expected) actual = render.render_node(node) self.assertEqual(actual, expected)
def render_text(module_text): tree = parse(module_text) recast_docstrings(tree) adjust_lines(tree) return render.render_node(tree)