是否存在获取所有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
相关产品推荐
相关产品推荐

