如何为Pydantic嵌套类配置Hypothesis策略实现字段间约束

针对嵌套Pydantic模型生成测试数据时，跨字段的自定义语义约束（如end > start）无法通过点号传参、测试内临时修正的方式优雅实现，以下是可直接落地的方案：

推荐方案：分层定义composite策略 #

st.builds()仅识别传入模型的直接字段名，不支持点号分隔的嵌套路径写法，因此只需要从最内层模型开始，逐层定义带约束的生成策略即可：

from pydantic import BaseModel, Field, conint
from hypothesis import given, strategies as st

class Inner(BaseModel):
    start: conint(ge=0, le=9) = Field(description="starting value")
    end: conint(ge=0, le=9) = Field(description="it should be greater than start")


class Outer(BaseModel):
    inner: Inner = Field(description="I level class")

# 1. 先为内层Inner模型定义符合约束的生成策略
@st.composite
def valid_inner(draw):
    # 先生成start值，再基于start的取值范围生成合法的end值，无重试性能损耗
    start_val = draw(st.integers(min_value=0, max_value=8))
    end_val = draw(st.integers(min_value=start_val + 1, max_value=9))
    return Inner(start=start_val, end=end_val)

# 2. 外层模型直接传入内层的自定义策略作为字段生成规则
@given(st.builds(Outer, inner=valid_inner()))
def test_schema(schema):
    assert schema.inner.start < schema.inner.end, "end value must be greater than start"


test_schema()

这种写法和模型的层级结构完全对应，后续调整字段约束时只需要修改对应层级的策略，维护成本最低。

其他可选实现 #

如果你不想单独定义composite函数，也可以通过基础策略组合实现：

# 生成合法的(start, end)元组后直接映射为Inner实例
valid_inner = st.tuples(
    st.integers(min_value=0, max_value=9),
    st.integers(min_value=0, max_value=9)
).filter(lambda pair: pair[0] < pair[1]).map(lambda pair: Inner(start=pair[0], end=pair[1]))

@given(st.builds(Outer, inner=valid_inner))
def test_schema(schema):
    assert schema.inner.start < schema.inner.end

注意：filter方法在约束条件严苛时会触发Hypothesis的多次重试，生成数据的效率低于composite方案，仅适合约束宽松的场景。

你之前提到的测试内交换值的修正逻辑，完全可以放到composite策略内部执行，不需要写在测试用例里：

@st.composite
def valid_inner(draw):
    start_val = draw(st.integers(0, 9))
    end_val = draw(st.integers(0, 9))
    # 在这里直接做值修正，保证返回的实例永远符合约束
    if start_val >= end_val:
        start_val, end_val = end_val, start_val
    return Inner(start=start_val, end=end_val)

之前尝试失效的原因 #

• 点号分隔传参（如inner.start）不符合st.builds()的参数规则：该方法接收的关键字参数必须和模型的直接属性名一一对应，不会自动解析嵌套路径，因此inner字段只能传入返回Inner类型实例的策略，无法直接给内层字段传生成规则。
• 测试内临时修正的逻辑没有封装为策略：Hypothesis的策略本质上是“返回合法测试数据的可调用对象”，只要把修正逻辑放到composite函数里，就能被Hypothesis识别为合法的自定义策略。

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