为何针对单成员枚举的不等性检查类型收窄仅对部分联合类型生效？

问题重现 #

示例1：str | E 无报错 #

from dataclasses import dataclass
from uuid import UUID
from enum import Enum

class E(Enum):
    m = "m"

@dataclass
class A:
    e: str | E

a = A(E.m)

def f(x: str) -> None:
    print(x)

if a.e != E.m:
    f(a.e)

在--strict模式下运行Mypy未报错，它能正确推断：若a.e != E.m成立，则a.e的类型只能是str。

示例2：UUID | E 报错 #

修改类型注解和函数定义：

@dataclass
class A:
    e: UUID | E

def f(x: UUID) -> None:
    print(x)

此时Mypy抛出类型错误：

bug.py:18: error: Argument 1 to "f" has incompatible type "UUID | E"; expected "UUID"  [arg-type]
Found 1 error in 1 file (checked 1 source file)

原因分析 #

这种差异是Mypy类型缩小规则的合理逻辑，并非bug，核心在于枚举类型与不同比较对象的交互处理：

• str与枚举的特殊适配：枚举E的成员m取值为字符串"m"，Mypy知晓枚举的__eq__方法会自动比较成员与其对应的值。因此当判断a.e != E.m时，Mypy能确定：如果a.e是E类型，那它必然是E.m，此时判断结果为假，不会进入分支；进入分支的a.e只能是str类型。
• UUID与枚举无特殊适配：UUID和E是完全无关的类型，且UUID并非E成员的取值类型。Mypy的类型系统未建模这种跨类型比较的确定性（尽管实际运行中UUID实例永远不等于E.m），因此无法通过a.e != E.m的判断将类型从UUID | E缩小为UUID。

解决办法 #

要让Mypy正确识别分支内的类型，可采用两种方式：

• 显式检查类型：

  if not isinstance(a.e, E):
      f(a.e)
  

• 使用类型断言（需确保逻辑正确性）：

  from typing import cast
  
  if a.e != E.m:
      f(cast(UUID, a.e))
  

内容的提问来源于stack exchange，提问作者sh-at-cs