Python Optional类型参数断言非空后mypy仍报类型错误的解决方法

mypy的自动类型收窄存在明确的识别边界：只有当它能直接追踪到对某个独立变量的非空校验逻辑时，才会将变量类型从 Optional[float] 收窄为实际的 float 类型。
原有代码先将多个独立的Optional变量打包为临时元组 reqs，再通过遍历元组的方式批量执行非空检查，mypy无法将「元组内元素非空」的校验结果反向映射回构成元组的原始独立变量。即使代码在运行时通过assert保证了非空逻辑，静态检查阶段依然会判定Xt、Xd等变量存在为None的可能，因此触发「不支持float和None类型做**运算」的报错。

方案1（推荐）：替换为可被mypy识别的非空校验逻辑 #

这种方式同时保留运行时校验能力和静态类型检查的准确性，无额外隐藏风险。
将原有的「打包成元组再循环检查」的逻辑，替换为两种mypy可精准识别的校验形式：

• 逐行对参与计算的变量做独立非空断言：

  if use_wcog:
      assert Xt is not None
      assert Xd is not None
      assert Xw is not None
      res = ((Xt**2 + Xw**2) / Xd)**2
  else:
      assert Xs is not None
      assert Xd is not None
      res = (Xs**2 / Xd)**2
  

• 将所有非空判断直接作为参数传入all()，不做中间元组打包：

  if use_wcog:
      assert all([
          Xt is not None,
          Xd is not None,
          Xw is not None,
      ])
      res = ((Xt**2 + Xw**2) / Xd)**2
  else:
      assert all([
          Xs is not None,
          Xd is not None,
      ])
      res = (Xs**2 / Xd)**2
  

两种写法下，mypy均可准确识别每个变量的非空状态，自动将类型收窄为float，不会再触发类型不匹配的报错。

方案2：使用mypy非空断言标记 #

如果可以百分百确认代码运行到计算环节时，参与计算的变量一定不会为None，可在对应变量后添加mypy专属的非空断言标记（英文感叹号!），直接告知检查工具该变量不存在None的可能：

if use_wcog:
    reqs = (Xt, Xd, Xw)
    assert all([_ is not None for _ in reqs])
    res = (((Xt!)**2 + (Xw!)**2) / (Xd!))**2
else:
    reqs = (Xs, Xd)
    assert all([_ is not None for _ in reqs])
    res = (((Xs!)**2) / (Xd!))**2

⚠️ 注意：该标记仅对mypy静态检查生效，不会在Python运行时修改任何逻辑。若实际运行中变量为None，不会提前抛出断言错误，而是会在计算环节触发生硬的TypeError，问题排查成本更高。

方案3：通过cast手动指定变量类型 #

如果不想改动原有的批量检查逻辑，可在非空校验完成后，通过typing.cast手动告知mypy对应变量的实际非空类型：

from typing import Optional, cast

if use_wcog:
    reqs = (Xt, Xd, Xw)
    assert all([_ is not None for _ in reqs])
    Xt = cast(float, Xt)
    Xd = cast(float, Xd)
    Xw = cast(float, Xw)
    res = ((Xt**2 + Xw**2) / Xd)**2

同样需要注意：cast仅为静态类型层面的标记，运行时不执行任何校验，可靠性低于显式的is not None断言。

优先选择第一种方案，它保留了运行时的非空校验能力，当传入参数不符合要求时会抛出定位清晰的AssertionError，同时完全兼容mypy的检查逻辑，无额外使用成本。 #

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