Fix single item enum match type exhaustion

python · Feb 29, 2024 · 14699cb · 14699cb
1 parent 02c50bc
commit 14699cb
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/mypy/checkpattern.py b/mypy/checkpattern.py
@@ -202,7 +202,7 @@ def visit_value_pattern(self, o: ValuePattern) -> PatternType:
         typ = self.chk.expr_checker.accept(o.expr)
         typ = coerce_to_literal(typ)
         narrowed_type, rest_type = self.chk.conditional_types_with_intersection(
-            current_type, [get_type_range(typ)], o, default=current_type
+            current_type, [get_type_range(typ)], o, default=get_type_range(typ).item
         )
         if not isinstance(get_proper_type(narrowed_type), (LiteralType, UninhabitedType)):
             return PatternType(narrowed_type, UnionType.make_union([narrowed_type, rest_type]), {})

diff --git a/test-data/unit/check-python310.test b/test-data/unit/check-python310.test
@@ -1369,6 +1369,35 @@ match m3:
         reveal_type(m3)  # N: Revealed type is "Tuple[Union[builtins.int, builtins.str]]"
 [builtins fixtures/tuple.pyi]
 
+[case testMatchEnumMultipleChoices]
+from enum import Enum
+class Medal(Enum):
+    gold = 1
+    silver = 1
+
+def f(m: Medal) -> None:
+    match m:
+        case Medal.gold:
+            reveal_type(m)  # N: Revealed type is "Literal[__main__.Medal.gold]"
+        case Medal.silver:
+            reveal_type(m)  # N: Revealed type is "Literal[__main__.Medal.silver]"
+        case _:
+            reveal_type(m)
+[builtins fixtures/bool.pyi]
+
+[case testMatchEnumSingleChoice]
+from enum import Enum
+class Medal(Enum):
+    gold = 1
+
+def f(m: Medal) -> None:
+    match m:
+        case Medal.gold:
+            reveal_type(m)  # N: Revealed type is "Literal[__main__.Medal.gold]"
+        case _:
+            reveal_type(m)
+[builtins fixtures/bool.pyi]
+
 [case testMatchLiteralPatternEnumNegativeNarrowing]
 from enum import Enum
 class Medal(Enum):