test: isolate tools for C tests

Author: mbdevpl

test/common.py

@@ -2,7 +2,6 @@
 
 import collections.abc
 import datetime
-import io
 import itertools
 import os
 import pathlib
@@ -14,7 +13,6 @@
 
 import horast
 import numpy as np
-import pycparser.c_ast
 # import static_typing as st
 import typed_ast.ast3
 import typed_astunparse
@@ -116,16 +114,6 @@ def make_tmp_folder(sub_path: pathlib.Path, input_path: pathlib.Path) -> pathlib
     return output_dir
 
 
-def c_ast_dump(node: pycparser.c_ast.Node) -> str:
-    io_ = io.StringIO()
-    node.show(io_, attrnames=True, nodenames=True, showcoord=True)
-    return io_.getvalue()
-
-
-def basic_check_c_ast(case: unittest.TestCase, path, c_tree, **kwargs):
-    basic_check_ast(case, path, c_tree, pycparser.c_ast.FileAST, '.yaml', c_ast_dump, **kwargs)
-
-
 def basic_check_cpp_code(case: unittest.TestCase, path, code, **kwargs):
     basic_check_code(case, path, code, 'cpp14', **kwargs)
 

test/test_c.py

@@ -16,7 +16,11 @@
     pass
 from transpyle.general import AstGeneralizer, CodeReader, Parser
 
-from .common import basic_check_c_ast, basic_check_python_ast, execute_on_language_examples
+from .common import basic_check_python_ast, execute_on_language_examples
+try:
+    from .tools_c import basic_check_c_ast
+except ImportError:
+    pass
 
 _LOG = logging.getLogger(__name__)
 

test/tools_c.py

@@ -0,0 +1,18 @@
+"""Tools for testing C language support."""
+
+import io
+import unittest
+
+import pycparser.c_ast
+
+from .common import basic_check_ast
+
+
+def c_ast_dump(node: pycparser.c_ast.Node) -> str:
+    io_ = io.StringIO()
+    node.show(io_, attrnames=True, nodenames=True, showcoord=True)
+    return io_.getvalue()
+
+
+def basic_check_c_ast(case: unittest.TestCase, path, c_tree, **kwargs):
+    basic_check_ast(case, path, c_tree, pycparser.c_ast.FileAST, '.yaml', c_ast_dump, **kwargs)