首页
/ Hypothesis项目中使用策略限制嵌套数据类中的浮点数值范围

Hypothesis项目中使用策略限制嵌套数据类中的浮点数值范围

2025-05-28 18:21:43作者:余洋婵Anita

在软件开发过程中,我们经常需要生成测试数据来验证代码的正确性。Hypothesis是一个强大的Python测试库,它能够自动生成各种测试用例。本文将重点介绍如何在使用Hypothesis生成嵌套数据类实例时,对浮点数值进行范围限制。

问题背景

当我们需要测试一个处理嵌套数据结构的系统时,Hypothesis的from_type策略非常有用,它可以自动为复杂的数据类生成实例。然而,有时我们需要对这些生成的值施加额外的约束条件。

例如,当这些数据最终需要被序列化为JSON格式时,JSON规范不支持NaN(非数字)和Infinity(无穷大)这样的特殊浮点数值。因此,我们需要确保生成的测试数据中不包含这些特殊值。

解决方案

Hypothesis提供了一个优雅的解决方案:通过register_type_strategy函数注册全局类型策略。对于浮点数类型,我们可以这样设置:

from hypothesis import strategies as st

st.register_type_strategy(
    float,
    st.floats(
        allow_nan=False,
        allow_infinity=False,
        allow_subnormal=False
    )
)

这段代码注册了一个全局的浮点数生成策略,它会应用于所有浮点数字段,包括:

  • 禁止生成NaN值
  • 禁止生成无穷大值
  • 禁止生成非正规数(极小值)

实际应用示例

假设我们有以下嵌套的数据类结构:

from dataclasses import dataclass

@dataclass
class Child:
    f1: float
    f2: float | None

@dataclass
class Parent:
    child: Child

在注册了上述策略后,使用from_type(Parent).example()生成的实例将自动遵守这些约束条件,确保所有浮点数字段都是有效的JSON可序列化值。

注意事项

  1. 策略作用范围:注册的类型策略是全局性的,会影响项目中所有使用该类型的策略生成。

  2. 优先级:如果为特定字段显式定义了策略,它将覆盖全局注册的策略。

  3. 可重现性:为了确保测试的可重现性,建议在测试开始时设置随机种子,或者使用Hypothesis提供的@reproduce_failure装饰器。

  4. 性能影响:限制浮点数范围可能会略微影响生成效率,因为需要过滤掉不符合条件的值。

扩展应用

这种技术不仅适用于浮点数,还可以应用于其他需要全局约束的类型。例如:

  • 限制字符串长度
  • 控制整数范围
  • 确保日期在合理范围内

通过合理使用类型策略注册,我们可以确保生成的测试数据既满足测试需求,又符合业务规则和系统约束条件。

总结

在Hypothesis测试框架中,通过register_type_strategy注册全局类型策略是一种强大而灵活的方法,可以确保生成的测试数据符合特定约束条件。这种方法特别适合处理嵌套数据结构中的类型约束问题,能够显著提高测试代码的健壮性和可维护性。

登录后查看全文
热门项目推荐
相关项目推荐

项目优选

收起
kernelkernel
deepin linux kernel
C
22
6
docsdocs
OpenHarmony documentation | OpenHarmony开发者文档
Dockerfile
203
2.18 K
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
208
285
pytorchpytorch
Ascend Extension for PyTorch
Python
62
94
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
977
575
nop-entropynop-entropy
Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
9
1
ops-mathops-math
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
550
84
openHiTLSopenHiTLS
旨在打造算法先进、性能卓越、高效敏捷、安全可靠的密码套件,通过轻量级、可剪裁的软件技术架构满足各行业不同场景的多样化要求,让密码技术应用更简单,同时探索后量子等先进算法创新实践,构建密码前沿技术底座!
C
1.02 K
399
communitycommunity
本项目是CANN开源社区的核心管理仓库,包含社区的治理章程、治理组织、通用操作指引及流程规范等基础信息
393
27
MateChatMateChat
前端智能化场景解决方案UI库,轻松构建你的AI应用,我们将持续完善更新,欢迎你的使用与建议。 官网地址:https://matechat.gitcode.com
1.2 K
133