[Python-checkins] [3.12] gh-105164: Detect annotations inside match blocks (GH-105177) (#105313)
JelleZijlstra
webhook-mailer at python.org
Mon Jun 5 09:42:38 EDT 2023
https://github.com/python/cpython/commit/6d0354167f8093183ae5f12205fd0cf991dbbd6a
commit: 6d0354167f8093183ae5f12205fd0cf991dbbd6a
branch: 3.12
author: Miss Islington (bot) <31488909+miss-islington at users.noreply.github.com>
committer: JelleZijlstra <jelle.zijlstra at gmail.com>
date: 2023-06-05T13:42:31Z
summary:
[3.12] gh-105164: Detect annotations inside match blocks (GH-105177) (#105313)
(cherry picked from commit 69d1245685cf95ddc678633e978a56673da64865)
Co-authored-by: Jelle Zijlstra <jelle.zijlstra at gmail.com>
files:
A Misc/NEWS.d/next/Core and Builtins/2023-05-31-19-35-22.gh-issue-105164.6Wajph.rst
M Lib/test/test_type_annotations.py
M Python/compile.c
diff --git a/Lib/test/test_type_annotations.py b/Lib/test/test_type_annotations.py
index 87f46c2ce8ce..3dbb35afcb62 100644
--- a/Lib/test/test_type_annotations.py
+++ b/Lib/test/test_type_annotations.py
@@ -1,4 +1,6 @@
+import textwrap
import unittest
+from test.support import run_code
class TypeAnnotationTests(unittest.TestCase):
@@ -101,3 +103,112 @@ class D(metaclass=C):
with self.assertRaises(AttributeError):
del D.__annotations__
self.assertEqual(D.__annotations__, {})
+
+
+class TestSetupAnnotations(unittest.TestCase):
+ def check(self, code: str):
+ code = textwrap.dedent(code)
+ for scope in ("module", "class"):
+ with self.subTest(scope=scope):
+ if scope == "class":
+ code = f"class C:\n{textwrap.indent(code, ' ')}"
+ ns = run_code(code)
+ if scope == "class":
+ annotations = ns["C"].__annotations__
+ else:
+ annotations = ns["__annotations__"]
+ self.assertEqual(annotations, {"x": int})
+
+ def test_top_level(self):
+ self.check("x: int = 1")
+
+ def test_blocks(self):
+ self.check("if True:\n x: int = 1")
+ self.check("""
+ while True:
+ x: int = 1
+ break
+ """)
+ self.check("""
+ while False:
+ pass
+ else:
+ x: int = 1
+ """)
+ self.check("""
+ for i in range(1):
+ x: int = 1
+ """)
+ self.check("""
+ for i in range(1):
+ pass
+ else:
+ x: int = 1
+ """)
+
+ def test_try(self):
+ self.check("""
+ try:
+ x: int = 1
+ except:
+ pass
+ """)
+ self.check("""
+ try:
+ pass
+ except:
+ pass
+ else:
+ x: int = 1
+ """)
+ self.check("""
+ try:
+ pass
+ except:
+ pass
+ finally:
+ x: int = 1
+ """)
+ self.check("""
+ try:
+ 1/0
+ except:
+ x: int = 1
+ """)
+
+ def test_try_star(self):
+ self.check("""
+ try:
+ x: int = 1
+ except* Exception:
+ pass
+ """)
+ self.check("""
+ try:
+ pass
+ except* Exception:
+ pass
+ else:
+ x: int = 1
+ """)
+ self.check("""
+ try:
+ pass
+ except* Exception:
+ pass
+ finally:
+ x: int = 1
+ """)
+ self.check("""
+ try:
+ 1/0
+ except* Exception:
+ x: int = 1
+ """)
+
+ def test_match(self):
+ self.check("""
+ match 0:
+ case 0:
+ x: int = 1
+ """)
diff --git a/Misc/NEWS.d/next/Core and Builtins/2023-05-31-19-35-22.gh-issue-105164.6Wajph.rst b/Misc/NEWS.d/next/Core and Builtins/2023-05-31-19-35-22.gh-issue-105164.6Wajph.rst
new file mode 100644
index 000000000000..7d3486c3b6e9
--- /dev/null
+++ b/Misc/NEWS.d/next/Core and Builtins/2023-05-31-19-35-22.gh-issue-105164.6Wajph.rst
@@ -0,0 +1,2 @@
+Ensure annotations are set up correctly if the only annotation in a block is
+within a :keyword:`match` block. Patch by Jelle Zijlstra.
diff --git a/Python/compile.c b/Python/compile.c
index f2314ae11c41..32eda4d407ea 100644
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -1422,8 +1422,18 @@ find_ann(asdl_stmt_seq *stmts)
find_ann(st->v.TryStar.finalbody) ||
find_ann(st->v.TryStar.orelse);
break;
+ case Match_kind:
+ for (j = 0; j < asdl_seq_LEN(st->v.Match.cases); j++) {
+ match_case_ty match_case = (match_case_ty)asdl_seq_GET(
+ st->v.Match.cases, j);
+ if (find_ann(match_case->body)) {
+ return true;
+ }
+ }
+ break;
default:
res = false;
+ break;
}
if (res) {
break;
More information about the Python-checkins
mailing list