Added extra test for unpacking assignment

agronholm · Mar 15, 2023 · 84331cd · 84331cd
1 parent 188e002
commit 84331cd
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/tests/dummymodule.py b/tests/dummymodule.py
@@ -202,3 +202,14 @@ def unpacking_assign() -> Tuple[int, str]:
     x: int
     x, y = (1, "foo")
     return x, y
+
+
+@typechecked
+def unpacking_assign_generator() -> Tuple[int, str]:
+    def genfunc():
+        yield 1
+        yield "foo"
+
+    x: int
+    x, y = genfunc()
+    return x, y
diff --git a/tests/test_instrumentation.py b/tests/test_instrumentation.py
@@ -214,3 +214,7 @@ def test_augmented_assign(dummymodule):
 
 def test_unpacking_assign(dummymodule):
     assert dummymodule.unpacking_assign() == (1, "foo")
+
+
+def test_unpacking_assign_from_generator(dummymodule):
+    assert dummymodule.unpacking_assign_generator() == (1, "foo")