Python泛型编程：TypeVar与Union兼容性解析

TypeVar约束与Union类型的核心冲突

在Python的类型提示系统中，TypeVar是实现泛型编程的强大工具，允许我们编写能够处理多种类型但保持类型一致性的函数或类。然而，当TypeVar被赋予显式约束时，它与Union类型（即A | B）的交互方式有时会出乎意料，导致类型检查器（如Pyright或Mypy）报错。

考虑一个函数f，它期望接收一个float或Fraction类型的值，并返回相同类型的值。我们可能会这样定义一个受约束的TypeVar：

from fractions import Fraction
from typing import TypeVar

# T 被约束为 float 或 Fraction
T = TypeVar("T", float, Fraction) 

def f(x: T) -> T:
    """
    期望接收一个 float 或 Fraction，并返回相同类型的值。
    """
    return x * 2

# 以下调用是合法的
print(f(1.0))          # T 被推断为 float
print(f(Fraction(1, 2))) # T 被推断为 Fraction

现在，假设我们有另一个函数g，它接受一个float或Fraction的联合类型，并尝试将其传递给f：

def g(x: float | Fraction) -> float | Fraction:
    """
    期望接收一个 float 或 Fraction 的联合类型。
    """
    # 当尝试将 x 传递给 f 时，Pyright 会报错
    return f(x) / 2 

类型检查器（如Pyright）会报告类似以下的错误：

Argument of type "float | Fraction" cannot be assigned to parameter "x" of type "T@f" in function "f"
Type "float | Fraction" is incompatible with constrained type variable "T"

这个错误揭示了核心问题：T = TypeVar("T", float, Fraction)的含义是，T在任何特定调用中，必须精确地是float或精确地是Fraction。它不是指T可以是float和Fraction的联合体。当x的类型是float | Fraction时，类型检查器无法确定x是float还是Fraction，因此它不能将其直接赋值给期望T（一个具体类型）的参数。

注意事项：float | int为何可能通过检查？

值得注意的是，如果我们将Fraction替换为内置的int类型，例如T = TypeVar("T", float, int)，那么f(x: float | int)可能不会报错。这并非因为float | int作为联合类型被TypeVar自动接受，而是由于Python类型系统对内置数值类型（特别是int到float的隐式类型提升）的特殊处理。在许多上下文中，int可以被视为float的子类型或可兼容类型。然而，对于Fraction或numpy.ndarray等非内置或更复杂的类型，这种隐式兼容性并不存在，从而暴露了TypeVar与Union之间更普遍的冲突。对于非内置类型，上述问题依然存在。

解决方案一：扩展TypeVar的约束集

解决上述问题的一种方法是明确告诉TypeVar，它除了可以是float或Fraction之外，也可以是float | Fraction这个联合类型本身。

from fractions import Fraction
from typing import TypeVar

# 将联合类型 float | Fraction 也添加到 TypeVar 的约束中
T_Extended = TypeVar("T_Extended", float, Fraction, float | Fraction)

def f_extended(x: T_Extended) -> T_Extended:
    """
    期望接收一个 float、Fraction 或它们的联合类型，并返回相同类型的值。
    """
    return x * 2

def g_fixed_1(x: float | Fraction) -> float | Fraction:
    """
    现在可以正确地将联合类型传递给 f_extended。
    """
    return f_extended(x) / 2 # 现在类型检查通过

在这个方案中，当x的类型是float | Fraction时，TypeVar T_Extended会被推断为float | Fraction。这意味着函数f_extended的输入和输出类型都将是float | Fraction。

适用场景： 当你需要TypeVar在某些情况下精确地代表其约束集中某个具体类型，而在另一些情况下精确地代表这些具体类型的联合时，此方法非常有用。它提供了对TypeVar推断结果的精确控制。

解决方案二：使用bound参数定义上界

另一种更灵活且在许多泛型场景中更常用的方法是使用TypeVar的bound参数。bound参数定义了类型变量的上限，意味着T可以是bound类型或其任何子类型。

from fractions import Fraction
from typing import Type