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)
