diff options
author | Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> | 2022-09-05 17:39:52 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-09-05 17:39:52 (GMT) |
commit | 6cc31af657a0aaf5a8ba0564b35124f1ca542287 (patch) | |
tree | 4b97e2e56a271e3f46b354722556a4db82811206 | |
parent | e13f49a0bc253f1b8939fd505547986183761768 (diff) | |
download | cpython-6cc31af657a0aaf5a8ba0564b35124f1ca542287.zip cpython-6cc31af657a0aaf5a8ba0564b35124f1ca542287.tar.gz cpython-6cc31af657a0aaf5a8ba0564b35124f1ca542287.tar.bz2 |
gh-92986: Fix ast.unparse when ImportFrom.level is None (GH-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
(GH-57645, GH-92987)
(cherry picked from commit 200c9a8da0e2b892c476807e986009c01327e781)
Co-authored-by: Shantanu <12621235+hauntsaninja@users.noreply.github.com>
-rw-r--r-- | Lib/ast.py | 2 | ||||
-rw-r--r-- | Lib/test/test_unparse.py | 6 | ||||
-rw-r--r-- | Misc/NEWS.d/next/Library/2022-05-19-22-34-42.gh-issue-92986.e6uKxj.rst | 1 |
3 files changed, 8 insertions, 1 deletions
@@ -849,7 +849,7 @@ class _Unparser(NodeVisitor): def visit_ImportFrom(self, node): self.fill("from ") - self.write("." * node.level) + self.write("." * (node.level or 0)) if node.module: self.write(node.module) self.write(" import ") diff --git a/Lib/test/test_unparse.py b/Lib/test/test_unparse.py index 33e1149..6e38c5e 100644 --- a/Lib/test/test_unparse.py +++ b/Lib/test/test_unparse.py @@ -341,6 +341,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"', diff --git a/Misc/NEWS.d/next/Library/2022-05-19-22-34-42.gh-issue-92986.e6uKxj.rst b/Misc/NEWS.d/next/Library/2022-05-19-22-34-42.gh-issue-92986.e6uKxj.rst new file mode 100644 index 0000000..691c0dd --- /dev/null +++ b/Misc/NEWS.d/next/Library/2022-05-19-22-34-42.gh-issue-92986.e6uKxj.rst @@ -0,0 +1 @@ +Fix :func:`ast.unparse` when ``ImportFrom.level`` is None |