allow redefinition of functions across match arms (#772)

PyCQA · Apr 25, 2023 · f2671ff · f2671ff
1 parent e19886e
commit f2671ff
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/pyflakes/checker.py b/pyflakes/checker.py
@@ -29,8 +29,10 @@
 def getAlternatives(n):
     if isinstance(n, ast.If):
         return [n.body]
-    if isinstance(n, ast.Try):
+    elif isinstance(n, ast.Try):
         return [n.body + n.orelse] + [[hdl] for hdl in n.handlers]
+    elif sys.version_info >= (3, 10) and isinstance(n, ast.Match):
+        return [mc.body for mc in n.cases]
 
 
 FOR_TYPES = (ast.For, ast.AsyncFor)

diff --git a/pyflakes/test/test_match.py b/pyflakes/test/test_match.py
@@ -81,3 +81,14 @@ def test_match_double_star(self):
                 case {'foo': k1, **rest}:
                     print(f'{k1=} {rest=}')
         ''')
+
+    def test_defined_in_different_branches(self):
+        self.flakes('''
+            def f(x):
+                match x:
+                    case 1:
+                        def y(): pass
+                    case _:
+                        def y(): print(1)
+                return y
+        ''')