You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

是否存在获取所有NumPy类型的内置方法?单元测试需求问询

用NumPy内置方法自动获取所有类型用于单元测试

你需要编写单元测试验证一个仅接受numpy.uint8类型数组的函数,希望自动获取所有NumPy类型并排除uint8,替代手动罗列不支持类型的方式。

NumPy没有直接返回所有内置类型的单一函数,但可以通过np.sctypes结合补充部分特殊类型来实现需求:

import numpy as np

# 收集所有NumPy基础类型
all_numpy_types = set()
# np.sctypes包含各类基础类型分组:整数、无符号整数、浮点数等
for type_group in np.sctypes.values():
    all_numpy_types.update(type_group)
# 补充scctypes未包含的特殊类型
all_numpy_types.update([np.datetime64, np.timedelta64, np.void])

# 移除支持的uint8类型,得到所有不支持类型
not_supported_types = all_numpy_types - {np.uint8}

说明:

  • np.sctypes是一个字典,键包括int、uint、float、complex、bool、object、bytes、str,对应的值是该类别下的所有NumPy类型列表。
  • np.datetime64、np.timedelta64、np.void这几个类型不在np.sctypes的分组里,需要手动添加进去。
  • 这种方式无需手动维护类型列表,NumPy版本更新新增类型时,代码会自动包含新类型,比手动罗列更可靠。

内容的提问来源于stack exchange,提问作者Ivan

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.28 17:07:11