diff options
author | Shantanu <12621235+hauntsaninja@users.noreply.github.com> | 2022-09-05 17:14:50 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-09-05 17:14:50 (GMT) |
commit | 200c9a8da0e2b892c476807e986009c01327e781 (patch) | |
tree | f44c3b45c7e36e879a7f2bf6f613cfbdabc69936 /Lib/test/test_unparse.py | |
parent | 2c7d2e8d46164efb6e27a64081d8e949f6876515 (diff) | |
download | cpython-200c9a8da0e2b892c476807e986009c01327e781.zip cpython-200c9a8da0e2b892c476807e986009c01327e781.tar.gz cpython-200c9a8da0e2b892c476807e986009c01327e781.tar.bz2 |
gh-92986: Fix ast.unparse when ImportFrom.level is None (#92992)
This doesn't happen naturally, but is allowed by the ASDL and compiler.
We don't want to change ASDL for backward compatibility reasons
(#57645, #92987)
Diffstat (limited to 'Lib/test/test_unparse.py')
-rw-r--r-- | Lib/test/test_unparse.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/Lib/test/test_unparse.py b/Lib/test/test_unparse.py index 969aa16..f1f1dd5 100644 --- a/Lib/test/test_unparse.py +++ b/Lib/test/test_unparse.py @@ -422,6 +422,12 @@ class UnparseTestCase(ASTTestCase): def test_invalid_yield_from(self): self.check_invalid(ast.YieldFrom(value=None)) + def test_import_from_level_none(self): + tree = ast.ImportFrom(module='mod', names=[ast.alias(name='x')]) + self.assertEqual(ast.unparse(tree), "from mod import x") + tree = ast.ImportFrom(module='mod', names=[ast.alias(name='x')], level=None) + self.assertEqual(ast.unparse(tree), "from mod import x") + def test_docstrings(self): docstrings = ( 'this ends with double quote"', |