Python泛型协议中类型方差行为不符预期的疑问与验证

你遇到的这些问题其实涉及到Python泛型协议中类型方差和函数子类型化的细节，结合PEP 484（泛型）和PEP 544（协议）的规则，我来逐个帮你分析：

首先回顾你的测试代码：

from typing import TypeVar, Protocol

# Type variables
T = TypeVar('T')
T_co = TypeVar('T_co', covariant=True)
T_contra = TypeVar('T_contra', contravariant=True)

# Class hierarchy
class Animal: pass
class Dog(Animal): pass

# Protocols
class Feeder(Protocol[T]):
    def feed(self, animal: T) -> T: ...

class Adopter(Protocol[T_co]):
    def adopt(self) -> T_co: ...

class Walker(Protocol[T_contra]):
    def walk(self, animal: T_contra) -> None: ...

# Implementations
class AnimalFeeder:
    def feed(self, animal: Animal) -> Animal: ...

class DogFeeder:
    def feed(self, animal: Dog) -> Dog: ...

class AnimalAdopter:
    def adopt(self) -> Animal: ...

class DogAdopter:
    def adopt(self) -> Dog: ...

class AnimalWalker:
    def walk(self, animal: Animal) -> None: ...

class DogWalker:
    def walk(self, animal: Dog) -> None: ...

你的测试用例中出现了和预期不符的情况，下面逐个解答你的疑问：

问题1：为什么feeder3: Feeder[Animal] = DogFeeder()能通过类型检查？ #

你预期这应该失败，因为Feeder使用的是不变类型变量T，理论上Feeder[Animal]和Feeder[Dog]是完全独立的类型，不能互相赋值。但实际通过了，这其实是PyCharm类型检查器的局限，而非Python类型系统的规范：

我们拆解Feeder协议的方法签名：Feeder[T]要求的方法是feed(self, animal: T) -> T，而DogFeeder的方法是feed(self, animal: Dog) -> Dog。从函数子类型化的核心规则来看：

• 函数的参数类型是逆变的：如果要把函数A当作函数B的子类型，A的参数类型必须比B的参数类型更宽泛（或相等），这样才能安全接收B的参数。
• 函数的返回值类型是协变的：A的返回值类型必须比B的返回值类型更严格（或相等），这样返回的值能被B的调用者正确接收。

对于Feeder[Animal]，它要求的函数是(Animal) -> Animal，而DogFeeder的函数是(Dog) -> Dog：

• 参数部分：Dog比Animal更严格，不符合逆变要求——如果调用feeder3.feed(Animal())，DogFeeder的方法根本处理不了非Dog的Animal，这是类型不安全的。
• 返回值部分：Dog是Animal的子类，符合协变要求。

PyCharm可能只检查了返回值的协变，而忽略了参数的逆变冲突，导致错误地允许了这个赋值。按照Python类型系统规范，这个赋值本应该是不合法的，用mypy检查会标记为错误。

问题2：为什么walker2: Walker[Dog] = AnimalWalker()会失败？ #

Walker使用的是逆变类型变量T_contra，按照逆变规则：如果X是Y的子类，那么Walker[Y]是Walker[X]的子类。具体到你的场景：

• AnimalWalker的walk方法接受Animal类型，而Walker[Dog]要求的是接受Dog类型。因为Animal是Dog的超类，AnimalWalker的方法可以处理任何Animal，自然也能处理Dog，所以这个赋值是完全类型安全的。

PyCharm报错是因为它对逆变协议的支持不够完善，换用mypy检查的话，这个赋值会被判定为合法。

问题3：这些行为是Python类型系统的正确行为，还是PyCharm的限制？ #

结论是：这些是PyCharm类型检查器的局限性，而非Python类型系统的规范要求。

Python的类型系统（基于PEP 484和PEP 544）对类型方差的规则是明确的：

• 不变类型变量：泛型类型之间不存在子类型关系（比如Feeder[Animal]和Feeder[Dog]无关）
• 协变类型变量：如果X是Y的子类，那么Adopter[X]是Adopter[Y]的子类（你的adopter3符合这个规则，是正确的）
• 逆变类型变量：如果X是Y的子类，那么Walker[Y]是Walker[X]的子类（你的walker2应该合法，walker3应该不合法）

你看到的feeder3通过、walker2失败的情况，都是PyCharm在处理协议和方差时的不准确表现。建议用mypy重新检查你的代码，它会给出更符合规范的结果。

备注：内容来源于stack exchange，提问作者rednammoc