Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 1 addition & 7 deletions mypy/checkexpr.py
Original file line number Diff line number Diff line change
Expand Up @@ -4536,13 +4536,7 @@ def check_list_multiply(self, e: OpExpr) -> Type:

Type inference is special-cased for this common construct.
"""
right_type = self.accept(e.right)
if is_subtype(right_type, self.named_type("builtins.int")):
# Special case: [...] * <int value>. Use the type context of the
# OpExpr, since the multiplication does not affect the type.
left_type = self.accept(e.left, type_context=self.type_context[-1])
else:
left_type = self.accept(e.left)
left_type = self.accept(e.left, type_context=self.type_context[-1])
result, method_type = self.check_op("__mul__", left_type, e.right, e)
e.method_type = method_type
return result
Expand Down
6 changes: 6 additions & 0 deletions test-data/unit/check-inference-context.test
Original file line number Diff line number Diff line change
Expand Up @@ -847,6 +847,12 @@ a: List[int]
if int():
a = [None] * 3 # E: List item 0 has incompatible type "None"; expected "int"
a = [''] * 3 # E: List item 0 has incompatible type "str"; expected "int"

def walrus(n: int | None) -> None:
a: List[int] = [n] * (n := 1) # E: List item 0 has incompatible type "int | None"; expected "int"
n = None
b: List[object] = [(n := 1)] * n
reveal_type(n) # N: Revealed type is "builtins.int"
[builtins fixtures/list.pyi]

[case testUnionTypeContext]
Expand Down
Loading