diff options
Diffstat (limited to 'Demo/parser/test_unparse.py')
-rw-r--r-- | Demo/parser/test_unparse.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/Demo/parser/test_unparse.py b/Demo/parser/test_unparse.py index f4a3e66..7d85f16 100644 --- a/Demo/parser/test_unparse.py +++ b/Demo/parser/test_unparse.py @@ -64,6 +64,24 @@ class_decorator = """\ class Foo: pass """ +elif1 = """\ +if cond1: + suite1 +elif cond2: + suite2 +else: + suite3 +""" + +elif2 = """\ +if cond1: + suite1 +elif cond2: + suite2 +""" + + + class ASTTestCase(unittest.TestCase): def assertASTEqual(self, ast1, ast2): self.assertEqual(ast.dump(ast1), ast.dump(ast2)) @@ -159,6 +177,10 @@ class UnparseTestCase(ASTTestCase): def test_class_definition(self): self.check_roundtrip("class A(metaclass=type, *[], **{}): pass") + def test_elifs(self): + self.check_roundtrip(elif1) + self.check_roundtrip(elif2) + class DirectoryTestCase(ASTTestCase): """Test roundtrip behaviour on all files in Lib and Lib/test.""" |