处理动态返回类型：是否有现成的Python断言并类型转换函数？

现成第三方库实现 #

• assertpy：提供流畅的断言API，支持类型检查后返回原对象，兼顾运行时验证与类型提示：

  from assertpy import assert_that
  
  result = fsolve(...)
  result = assert_that(result).is_instance_of(numpy.ndarray).value
  # 后续可安全调用numpy数组方法，无类型错误
  

  处理可选类型非空断言场景：

  my_var: Optional[dict] = ...
  # 确认变量非空后
  my_var = assert_that(my_var).is_not_none().is_instance_of(dict).value
  my_var['abc']  # 无类型错误
  

• pytypes：其cast函数自带运行时类型检查，比标准库typing.cast多了验证环节：

  from pytypes import cast
  
  result = cast(numpy.ndarray, fsolve(...))
  # 类型不匹配时会抛出TypeError，类型检查器也能正确识别返回类型
  

无依赖的标准库替代方案 #

如果不想引入第三方库，标准库typing.cast可满足类型提示需求，但无运行时断言：

from typing import cast

result = cast(numpy.ndarray, fsolve(...))
# 类型检查器会认可result为numpy.ndarray，但运行时不做验证

若需要同时保留运行时检查和类型提示，你自己实现的assert_and_cast是非常规范的方案，还可优化为更健壮的版本：

from typing import Any, TypeVar, Type

T = TypeVar('T')

def assert_and_cast(obj: Any, typ: Type[T]) -> T:
    assert isinstance(obj, typ), f"Expected {typ.__name__}, got {type(obj).__name__}"
    return obj

该实现既在运行时做了类型校验，又能让类型检查器正确推断返回类型，完全匹配你的需求。

针对scipy特殊API的根源优化 #

对于scipy.optimize.fsolve这类返回类型随参数变化的函数，可通过重载函数类型提前为类型检查器提供信息，避免后续断言：

from typing import overload, Any, Tuple
import numpy as np
from scipy.optimize import fsolve

@overload
def fsolve(func: Any, x0: Any, full_output: bool = False, **kwargs: Any) -> np.ndarray: ...
@overload
def fsolve(func: Any, x0: Any, full_output: bool = True, **kwargs: Any) -> Tuple[np.ndarray, Any, int, str]: ...

# 调用时类型检查器会根据full_output参数自动推断返回类型
result = fsolve(..., full_output=False)  # 自动识别为np.ndarray

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