Use source_type.last_known_value when Instance. Fixes python#12923

CarlFK · Apr 25, 2023 · 06c1a61 · 06c1a61
1 parent 15fbd53
commit 06c1a61
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/mypy/checkexpr.py b/mypy/checkexpr.py
@@ -3918,6 +3918,9 @@ def visit_assert_type_expr(self, expr: AssertTypeExpr) -> Type:
             always_allow_any=True,
         )
         target_type = expr.type
+        if isinstance( source_type, mypy.types.Instance ) \
+                and source_type.last_known_value is not None:
+            source_type = source_type.last_known_value
         if not is_same_type(source_type, target_type):
             if not self.chk.in_checked_function():
                 self.msg.note(

diff --git a/test-data/unit/check-expressions.test b/test-data/unit/check-expressions.test
@@ -937,6 +937,8 @@ reveal_type(returned)  # N: Revealed type is "builtins.int"
 assert_type(a, str)  # E: Expression is of type "int", not "str"
 assert_type(a, Any)  # E: Expression is of type "int", not "Any"
 assert_type(a, Literal[1])  # E: Expression is of type "int", not "Literal[1]"
+assert_type(42, Literal[42])
+assert_type(42, int)  # E: Expression is of type "Literal[42]", not "int"
 [builtins fixtures/tuple.pyi]
 
 [case testAssertTypeGeneric]