如何在辅助函数场景下正确使用mypy进行类型检查

问题场景 #

下面是一段带类型提示的Python代码：

from typing import Literal

def func1(x: int):
    """Perform some work on integers."""
    if not isinstance(x, int):
        raise ValueError("x should be an integer.")
    
    # Do some stuff.

def func2(x: float):
    """Perform some work on floats."""
    if not isinstance(x, float):
        raise ValueError("x should be a float.")
    
    # Do some stuff.

def my_function(x: int | float, method: Literal[1, 2]):
    """Perform some work using one of two methods."""

    match method:
        case 1:
            result = func1(x)
        case 2:
            result = func2(x)
        case _:
            raise ValueError(f"Invalid method: {method}.")

    # Do some more work on result.

    return result

这段代码中，my_function根据method参数选择调用func1或func2，但mypy会报错，提示x的int | float类型与func1要求的int、func2要求的float类型不兼容。另外需要明确：不想通过x的类型自动选择函数，因为用户很容易混淆整数和浮点数格式（比如输入1而非1.0），必须让用户通过method明确指定处理逻辑。

推荐解决方式 #

1. 用类型断言或忽略类型检查 #

因为我们已经通过method参数明确了要调用的函数，且func1、func2自身带有运行时类型校验，所以可以直接告诉mypy当前x的类型符合函数要求，有两种实现方式：

方式一：使用type: ignore注释

直接忽略mypy的类型报错，因为运行时func1/func2会自行校验：

def my_function(x: int | float, method: Literal[1, 2]):
    """Perform some work using one of two methods."""

    match method:
        case 1:
            result = func1(x)  # type: ignore[arg-type]
        case 2:
            result = func2(x)  # type: ignore[arg-type]
        case _:
            raise ValueError(f"Invalid method: {method}.")

    # Do some more work on result.

    return result

方式二：使用cast类型转换

如果想让类型检查更严谨，可以用typing.cast明确告知mypyx的具体类型：

from typing import Literal, cast

# ... 保留func1、func2定义 ...

def my_function(x: int | float, method: Literal[1, 2]):
    """Perform some work using one of two methods."""

    match method:
        case 1:
            result = func1(cast(int, x))
        case 2:
            result = func2(cast(float, x))
        case _:
            raise ValueError(f"Invalid method: {method}.")

    # Do some more work on result.

    return result

可选优化：自动转换整数型浮点数

如果希望兼容用户输入1.0却选择method=1的场景，可以在调用func1前自动转换类型，同时保留func1的校验：

case 1:
    if isinstance(x, float) and x.is_integer():
        x = int(x)
    result = func1(cast(int, x))

2. 是否需要在my_function中重复类型检查？ #

完全不需要。理由如下：

• func1和func2已经内置了完整的运行时类型校验逻辑，即使my_function传入了不符合要求的类型，这两个函数会直接抛出ValueError，无需额外重复校验。
• 重复校验只会增加冗余代码，不会带来额外的安全性或可读性提升。

关于不按x类型自动选函数的说明 #

通过method参数让用户明确选择处理逻辑，能避免因用户输入格式混淆（比如把1.0写成1）导致的逻辑错误，让用户的意图更清晰，这比自动推断类型更可靠。

内容的提问来源于stack exchange，提问作者Biblot