Add support for setting enum members to "..."

Unblock python/typeshed#11299
JelleZijlstra · Jan 22, 2024 · 236b90b · 236b90b
1 parent 75b68fa
commit 236b90b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/mypy/stubtest.py b/mypy/stubtest.py
@@ -1093,6 +1093,9 @@ def verify_var(
             runtime_type = get_mypy_type_of_runtime_value(runtime.value)
             if runtime_type is not None and is_subtype_helper(runtime_type, stub.type):
                 should_error = False
+            # We always allow setting the stub value to ...
+            if isinstance(stub.type, mypy.types.Instance) and stub.type.type.fullname == "builtins.ellipsis":
+                should_error = False
 
         if should_error:
             yield Error(

diff --git a/mypy/test/teststubtest.py b/mypy/test/teststubtest.py
@@ -1139,6 +1139,25 @@ def baz(x=Flags3(0)): pass
             """,
             error=None,
         )
+        yield Case(
+            runtime="""
+            import enum
+            class SomeObject: ...
+
+            class WeirdEnum(enum.Enum):
+                a = SomeObject()
+                b = SomeObject()
+            """,
+            stub="""
+            import enum
+            class SomeObject: ...
+            class WeirdEnum(enum.Enum):
+                _value_: SomeObject
+                a = ...
+                b = ...
+            """,
+            error=None,
+        )
         yield Case(
             stub="""
             class Flags4(enum.Flag):