From c30573e7b95eef9d057ff42ebfd326438dac3c42 Mon Sep 17 00:00:00 2001 From: Ivan Levkivskyi Date: Mon, 27 Jan 2025 10:58:16 +0000 Subject: [PATCH] Fix literal context for ternary expressions (for real) (#18545) I am not waiting for review as the fix is obvious. The only annoying thing is that we had an exact test as in the repro but it passed accidentally because we use builtins fixtures. --- mypy/checker.py | 2 +- test-data/unit/check-literal.test | 2 +- test-data/unit/fixtures/primitives.pyi | 1 + 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/mypy/checker.py b/mypy/checker.py index 47b08b683e36f..c9495ca31842b 100644 --- a/mypy/checker.py +++ b/mypy/checker.py @@ -4372,7 +4372,7 @@ def check_simple_assignment( if ( isinstance(get_proper_type(lvalue_type), UnionType) # Skip literal types, as they have special logic (for better errors). - and not isinstance(get_proper_type(rvalue_type), LiteralType) + and not is_literal_type_like(rvalue_type) and not self.simple_rvalue(rvalue) ): # Try re-inferring r.h.s. in empty context, and use that if it diff --git a/test-data/unit/check-literal.test b/test-data/unit/check-literal.test index 296956334d205..246ce455c576a 100644 --- a/test-data/unit/check-literal.test +++ b/test-data/unit/check-literal.test @@ -2984,7 +2984,7 @@ class C(Base): sep = "a" if int() else "b" reveal_type(sep) # N: Revealed type is "Union[Literal['a'], Literal['b']]" return super().feed_data(sep) -[builtins fixtures/tuple.pyi] +[builtins fixtures/primitives.pyi] [case testLiteralInsideAType] from typing_extensions import Literal diff --git a/test-data/unit/fixtures/primitives.pyi b/test-data/unit/fixtures/primitives.pyi index fc220a4e2ee0c..2f8623c79b9ff 100644 --- a/test-data/unit/fixtures/primitives.pyi +++ b/test-data/unit/fixtures/primitives.pyi @@ -19,6 +19,7 @@ class int: def __init__(self, x: object = ..., base: int = ...) -> None: pass def __add__(self, i: int) -> int: pass def __rmul__(self, x: int) -> int: pass + def __bool__(self) -> bool: pass class float: def __float__(self) -> float: pass def __add__(self, x: float) -> float: pass