Fix inference from nested unpacking of iterables (#1311)

1 files changed, 42 insertions, 0 deletions

diff --git a/tests/unittest_protocols.py b/tests/unittest_protocols.py
index 69e50f11..9a477d02 100644
--- a/tests/unittest_protocols.py
+++ b/tests/unittest_protocols.py
@@ -70,6 +70,48 @@ class ProtocolTests(unittest.TestCase):
         for2_assnode = next(assign_stmts[1].nodes_of_class(nodes.AssignName))
         self.assertRaises(InferenceError, list, for2_assnode.assigned_stmts())
 
+    def test_assigned_stmts_nested_for_tuple(self) -> None:
+        assign_stmts = extract_node(
+            """
+        for a, (b, c) in [(1, (2, 3))]:  #@
+          pass
+        """
+        )
+
+        assign_nodes = assign_stmts.nodes_of_class(nodes.AssignName)
+
+        for1_assnode = next(assign_nodes)
+        assigned = list(for1_assnode.assigned_stmts())
+        self.assertConstNodesEqual([1], assigned)
+
+        for2_assnode = next(assign_nodes)
+        assigned2 = list(for2_assnode.assigned_stmts())
+        self.assertConstNodesEqual([2], assigned2)
+
+    def test_assigned_stmts_nested_for_dict(self) -> None:
+        assign_stmts = extract_node(
+            """
+        for a, (b, c) in {1: ("a", str), 2: ("b", bytes)}.items():  #@
+            pass
+        """
+        )
+        assign_nodes = assign_stmts.nodes_of_class(nodes.AssignName)
+
+        # assigned: [1, 2]
+        for1_assnode = next(assign_nodes)
+        assigned = list(for1_assnode.assigned_stmts())
+        self.assertConstNodesEqual([1, 2], assigned)
+
+        # assigned2: ["a", "b"]
+        for2_assnode = next(assign_nodes)
+        assigned2 = list(for2_assnode.assigned_stmts())
+        self.assertConstNodesEqual(["a", "b"], assigned2)
+
+        # assigned3: [str, bytes]
+        for3_assnode = next(assign_nodes)
+        assigned3 = list(for3_assnode.assigned_stmts())
+        self.assertNameNodesEqual(["str", "bytes"], assigned3)
+
     def test_assigned_stmts_starred_for(self) -> None:
         assign_stmts = extract_node(
             """