more tests

python · elazarg · Sep 3, 2016 · Sep 18, 2016 · Sep 18, 2016 · Sep 18, 2016
commit 6bb5519f3493cebc209d3cea2997e5f143b39dd2
diff --git a/mypy/checker.py b/mypy/checker.py
@@ -1177,7 +1177,7 @@ def check_multi_assign_from_union(self, lvalues: List[Expression], rvalue: Expre
                                         undefined_rvalue=True,
                                         infer_lvalue_type=infer_lvalue_type)
                 for t, lv in zip(transposed, lvalues):
-                    t.append(self.type_map[lv])
+                    t.append(self.type_map.get(lv, AnyType()))
         union_types = tuple(join_type_list(col) for col in transposed)
         for expr, items in assignments.items():
             types, declared_types = zip(*items)

diff --git a/test-data/unit/check-unions.test b/test-data/unit/check-unions.test
@@ -186,7 +186,66 @@ reveal_type(obj)  # E: Revealed type is 'builtins.int'
 
 [builtins fixtures/list.pyi]
 
--- [case testUnionListInit]
--- from typing import Union, List
--- a = []  # type: Union[List[int], List[str]]
--- [builtins fixtures/list.pyi]
+[case testUnionMultiassignAlreadyDeclared]
+from typing import Union, Tuple
+
+a = None  # type: Union[Tuple[int, int], Tuple[int, float]]
+a1 = None  # type: object
+a2 = None  # type: int
+
+(a1, a2) = a  # E: Incompatible types in assignment (expression has type "float", variable has type "int")
+
+b = None  # type: Union[Tuple[float, int], Tuple[int, int]]
+b1 = None  # type: object
+b2 = None  # type: int
+
+(b1, b2) = a  # E: Incompatible types in assignment (expression has type "float", variable has type "int")
+
+c = None  # type: Union[Tuple[int, int], Tuple[int, int]]
+c1 = None  # type: object
+c2 = None  # type: int
+
+(c1, c2) = c
+reveal_type(c1)  # E: Revealed type is 'builtins.int'
+reveal_type(c2)  # E: Revealed type is 'builtins.int'
+
+d = None  # type: Union[Tuple[int, int], Tuple[int, float]]
+d1 = None  # type: object
+
+(d1, d2) = d
+reveal_type(d1)  # E: Revealed type is 'builtins.int'
+reveal_type(d2)  # E: Revealed type is 'builtins.float'
+
+[case testUnionMultiassignIndexed]
+from typing import Union, Tuple, List
+
+class B:
+    x = None  # type: object
+
+x = None  # type: List[int]
+b = None  # type: B
+
+a = None  # type: Union[Tuple[int, int], Tuple[int, object]]
+(x[0], b.x) = a
+
+# I don't know why is it incomplete type
+reveal_type(x[0])  # E: Revealed type is 'builtins.int*'
+reveal_type(b.x)  # E: Revealed type is 'builtins.object'
+
+[builtins fixtures/list.pyi]
+
+[case testUnionMultiassignPacked]
+from typing import Union, Tuple, List
+
+a = None  # type: Union[Tuple[int, int, int], Tuple[int, int, str]]
+a1 = None  # type: int
+a2 = None  # type: object
+--FIX: allow proper rebinding of packed
+xs = None  # type: List[int]
+(a1, *xs, a2) = a 
+
+reveal_type(a1)  # E: Revealed type is 'builtins.int'
+reveal_type(xs)  # E: Revealed type is 'builtins.list[builtins.int]'
+reveal_type(a2)  # E: Revealed type is 'builtins.int'
+
+[builtins fixtures/list.pyi]
diff --git a/test-data/unit/fixtures/list.pyi b/test-data/unit/fixtures/list.pyi
@@ -19,6 +19,7 @@ class list(Iterable[T], Generic[T]):
     def __add__(self, x: list[T]) -> list[T]: pass
     def __mul__(self, x: int) -> list[T]: pass
     def __getitem__(self, x: int) -> T: pass
+    def __setitem__(self, x: int, v: T) -> None: pass
     def append(self, x: T) -> None: pass
     def extend(self, x: Iterable[T]) -> None: pass