Fix if (yield) bug

80c4d9a5 · Patrick Maupin · 810c2c2c · 80c4d9a5 · 80c4d9a5
Kaydet (Commit) 80c4d9a5 authored May 25, 2015 tarafından Patrick Maupin
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

op_util.py astor/op_util.py +2 -2

test_code_gen.py tests/test_code_gen.py +6 -0

No files found.
--- a/astor/op_util.py
+++ b/astor/op_util.py
@@ -19,10 +19,10 @@ op_data = """
          Assign                1
       AugAssign                0
            Expr                0
-              If                0
+           Yield                1
+              If                1
             For                0
           While                0
-           Yield                1
          Return                1

           Slice                1

--- a/tests/test_code_gen.py
+++ b/tests/test_code_gen.py
@@ -198,6 +198,12 @@ class CodegenTestCase(unittest.TestCase):
        self.assertAstEqual(source)
        source = "[(yield)]"
        self.assertAstEqual(source)
+        source = "if (yield): pass"
+        self.assertAstEqual(source)
+        source = "if (yield from foo): pass"
+        self.assertAstEqualIfAtLeastVersion(source, (3, 3))
+        source = "(yield from (a, b))"
+        self.assertAstEqualIfAtLeastVersion(source, (3, 3))

    def test_with(self):
        source = """