def test_greedy_decompose(fname, mode, type_comments, feature_version):
ast_tree = astutils.ast_parse(fname,
mode=mode,
type_comments=type_comments,
feature_version=feature_version)
sourcecode = _read(fname)
ast_heap = astutils.ast2heap(ast_tree, source=sourcecode)
sub_heaps = astutils.greedy_decompose(ast_heap)
def test_ast2dict_dict2ast(fname, mode, type_comments, feature_version):
ast_tree1 = astutils.ast_parse(fname,
mode=mode,
type_comments=type_comments,
feature_version=feature_version)
ast_dict = astutils.ast2dict(ast_tree1)
ast_tree2 = astutils.dict2ast(ast_dict)
ast_dict2 = astutils.ast2dict(ast_tree2)
assert _check_instances(ast_dict, ast_dict2)
def test_ast_unparse(fname, mode, type_comments, feature_version):
ast_tree = astutils.ast_parse(fname,
mode=mode,
type_comments=type_comments,
feature_version=feature_version)
if sys.version_info < (3, 9):
with pytest.raises(Exception):
astutils.ast_unparse(ast_tree)
else:
astutils.ast_unparse(ast_tree)
def test_ast2heap_heap2code(fname, mode, type_comments, feature_version):
ast_tree = astutils.ast_parse(fname,
mode=mode,
type_comments=type_comments,
feature_version=feature_version)
sourcecode = _read(fname)
ast_heap = astutils.ast2heap(ast_tree, source=sourcecode)
source_code_1 = astutils.heap2code(ast_heap)
assert _check_instances(sourcecode, source_code_1)
def test_ast2heap_heap2tokens(fname, mode, type_comments, feature_version):
ast_tree = astutils.ast_parse(fname,
mode=mode,
type_comments=type_comments,
feature_version=feature_version)
sourcecode = _read(fname)
ast_heap = astutils.ast2heap(ast_tree, source=sourcecode)
tokens = astutils.heap2tokens(ast_heap)
source_code_1 = "".join([token["tok"] for token in tokens])
assert _check_instances(sourcecode, source_code_1)
def test_ast_parse(fname, mode, type_comments, feature_version):
astutils.ast_parse(fname,
mode=mode,
type_comments=type_comments,
feature_version=feature_version)