如何让mypy识别类中可空属性传入numpy函数的类型错误

mypy默认不会自动推断实例变量的可选类型，你虽然在__init__方法里给参数a加了Optional标注，但没有显式声明self.a的类型，mypy会错误地认为self.a的类型由赋值语句固定，不会识别为可选类型，因此不会检测到传入None的风险。

• 方法1：显式声明实例变量的类型
  在类作用域下提前给a加上Optional类型标注，让mypy明确知道该属性可能为None：

from typing import Optional
import numpy as np

class TestClass():
    # 显式标注实例变量类型
    a: Optional[float]
    def __init__(self, a: Optional[float] = None):
        self.a = np.radians(a) if a is not None else None

    def test(self) -> None:
        # 此时mypy会正确识别self.a可能为None，抛出类型错误
        b = np.cos(self.a)

修改后再运行mypy，就会得到和直接写np.cos(None)完全一致的类型错误提示。

• 方法2：开启mypy的严格检查配置
  在mypy配置文件（如mypy.ini、pyproject.toml）中开启必要的检查项，确保实例变量的类型被正确校验：

# mypy.ini 配置示例
[mypy]
strict_optional = True
check_untyped_defs = True
disallow_incomplete_defs = True

临时运行也可以直接加命令行参数：mypy --strict-optional your_file.py

• 方法3：添加运行时断言（同时兼容类型检查和运行安全）
  如果业务逻辑要求调用test方法时self.a必须不为None，可以在方法开头添加断言，既可以让mypy自动窄化类型，也能在运行时提前抛出更明确的错误：

def test(self) -> None:
    assert self.a is not None, "调用test方法前必须先给a赋值"
    b = np.cos(self.a) # 此时mypy不会报错，因为已经保证a不为None

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