def test_roundtrip(self):
only_localizable = False
for name, example in EXAMPLES.items():
if ' with eol comments' in name or name.startswith('multiline '):
continue
data = {}
with self.subTest(name=name, example=example, data=data):
tree = typed_ast.ast3.parse(example)
code = typed_astunparse.unparse(tree)
complete_tree = parse(example)
data['complete_tree'] = complete_tree
complete_code = unparse(complete_tree)
data['complete_code'] = complete_code
self.assertGreaterEqual(len(complete_code.replace(' ', '')),
len(code.replace(' ', '')),
(complete_code, code))
reparsed_tree = typed_ast.ast3.parse(code)
tree_nodes = ast_to_list(tree, only_localizable)
reparsed_tree_nodes = ast_to_list(reparsed_tree,
only_localizable)
self.assertEqual(len(reparsed_tree_nodes), len(tree_nodes))
self.assertEqual(typed_ast.ast3.dump(reparsed_tree),
typed_ast.ast3.dump(tree))
reparsed_complete_tree = parse(complete_code)
complete_tree_nodes = ast_to_list(complete_tree,
only_localizable)
reparsed_complete_tree_nodes = ast_to_list(
reparsed_complete_tree, only_localizable)
self.assertEqual(len(reparsed_complete_tree_nodes),
len(complete_tree_nodes), complete_code)
self.assertEqual(
typed_ast.ast3.dump(reparsed_complete_tree),
typed_ast.ast3.dump(complete_tree),
'"""\n{}\n""" vs. original """\n{}\n"""'.format(
complete_code, example))
def test_unparse_eol_comment(self):
tree = typed_ast.ast3.If(
typed_ast.ast3.NameConstant(True),
[typed_ast.ast3.Pass(),
Comment(' hello world', eol=True)], [])
code = unparse(tree)
self.assertEqual(code.strip(), 'if True:\n pass # hello world')
def test_unparse_directive(self):
tree = typed_ast.ast3.If(
typed_ast.ast3.NameConstant(True),
[Directive('include<thisisnotcpp>'),
typed_ast.ast3.Pass()], [])
code = unparse(tree)
self.assertEqual(code.strip(),
'if True:\n #include<thisisnotcpp>\n pass')
def test_roundtrip_without_comments(self):
for name, example in EXAMPLES.items():
with self.subTest(name=name, example=example):
tree = typed_ast.ast3.parse(example)
code = unparse(tree)
reparsed_tree = typed_ast.ast3.parse(code)
self.assertEqual(typed_ast.ast3.dump(reparsed_tree),
typed_ast.ast3.dump(tree))
def test_roundtrip_openmp_pragma(self):
code = """# pragma: omp parallel for
# pragma: omp parallel do
# pragma: omp barrier"""
tree = parse(code)
for node in tree.body:
self.assertIsInstance(node, OpenMpPragma)
unparsed = unparse(tree).strip()
self.assertEqual(code, unparsed)
def test_roundtrip_openacc_pragma(self):
code = """# pragma: acc parallel
# pragma: acc end parallel
# pragma: acc parallel loop gang"""
tree = parse(code)
for node in tree.body:
self.assertIsInstance(node, OpenAccPragma)
unparsed = unparse(tree).strip()
self.assertEqual(code, unparsed)
def test_parse_type_comments(self):
for example in TYPE_COMMENT_EXAMPLES:
with self.subTest(nexample=example):
tree = parse(example)
self.assertIsNotNone(tree)
code = unparse(tree)
reparsed_tree = parse(code)
self.assertIsNotNone(reparsed_tree)
self.assertEqual(
typed_ast.ast3.dump(reparsed_tree),
typed_ast.ast3.dump(tree),
'"""\n{}\n""" vs. original """\n{}\n"""'.format(
code, example))
self.assertEqual(code.strip(), example)
def test_roundtrip_directive(self):
code = """#if spam
#else
#endif
#def
#undef
#ifdef
#ifndef"""
tree = parse(code)
for node in tree.body:
self.assertIsInstance(node, Directive)
self.assertIs(type(node), Directive)
unparsed = unparse(tree).strip()
self.assertEqual(code, unparsed)
def test_single_line(self):
code = """a = 1 # a equals one after this"""
tree = parse(code)
unparsed = unparse(tree).strip()
self.assertEqual(unparsed, code)
def test_roundtrip_comment(self):
code = """a = 1 # a equals one after this"""
tree = parse(code)
unparsed = unparse(tree).strip()
self.assertEqual(unparsed, code)
