首页
/ Hypothesis项目:如何从复合策略中获取单次测试用例

Hypothesis项目:如何从复合策略中获取单次测试用例

2025-05-29 10:33:22作者:郦嵘贵Just

在基于Hypothesis进行Python测试时,复合策略(composite strategy)是一个非常强大的工具,它允许开发者通过组合多个基础策略来创建复杂的测试数据生成器。然而,有时我们可能需要在非Hypothesis测试环境中使用这些策略生成的测试数据。

复合策略的基本用法

典型的复合策略使用@st.composite装饰器定义,例如生成IP网络和地址列表的策略:

from hypothesis import strategies as st

@st.composite
def list_of_ip_networks_and_addresses(draw):
    nets = draw(st.lists(ip_networks(), min_size=1, max_size=5))
    addresses = draw(st.lists(st.ip_addresses(v=4), min_size=1, max_size=5))
    return nets + addresses

在常规的Hypothesis测试中,这个策略会通过@given装饰器自动生成多个测试用例。

单次测试场景的需求

在某些情况下,开发者可能希望:

  1. 在非Hypothesis测试中使用策略生成的测试数据
  2. 只需获取一个随机示例而不需要完整的属性测试流程
  3. 避免为单次测试重新编写数据生成逻辑

解决方案:.example()方法

Hypothesis为策略对象提供了.example()方法,可以方便地获取单个随机示例:

def test_foo():
    # 获取单个随机示例
    lst_ips = list_of_ip_networks_and_addresses().example()
    # 使用该示例进行测试
    assert len(lst_ips) >= 2  # 根据策略定义,至少会有1个网络+1个地址

性能优化建议

如果需要频繁获取示例,可以考虑缓存策略对象以提高性能:

# 模块级别缓存策略对象
_ip_strategy = list_of_ip_networks_and_addresses()

def test_foo():
    lst_ips = _ip_strategy.example()
    # ...

注意事项

  1. .example()生成的测试数据不具有代表性,仅适合简单场景
  2. 对于关键测试,仍建议使用完整的属性测试(@given)
  3. 示例数据是随机生成的,每次运行可能不同

通过这种方法,开发者可以灵活地在各种测试场景中复用精心设计的复合策略,保持测试代码的一致性和可维护性。

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