Merge pull request #62 from berkerpeksag/interactive

Add support for ASTs compiled with single or eval options

astor/code_gen.py

@@ -713,6 +713,11 @@ class SourceGenerator(ExplicitNodeVisitor):
     def visit_Module(self, node):
         self.write(*node.body)
 
+    visit_Interactive = visit_Module
+
+    def visit_Expression(self, node):
+        self.visit(node.body)
+
     # Helper Nodes
 
     def visit_arg(self, node):

tests/test_code_gen.py

@@ -403,5 +403,12 @@ class CodegenTestCase(unittest.TestCase):
         """
         self.assertAstEqualIfAtLeastVersion(source, (3, 6))
 
+
+    def test_compile_types(self):
+        code = '(a + b + c) * (d + e + f)\n'
+        for mode in 'exec eval single'.split():
+            srcast = compile(code, 'dummy', mode, ast.PyCF_ONLY_AST)
+            dsttxt = astor.to_source(srcast)
+
 if __name__ == '__main__':
     unittest.main()