Pytest：使用假设生成总和为 1 的浮点数列表

Posted 2023-03-12

【中文标题】Pytest：使用假设生成总和为 1 的浮点数列表

【英文标题】：Pytest: generating list of floats summing up to 1 with hypothesis

【发布时间】：2021-01-09 17:40:42

【问题描述】：

对于单元测试，我想使用hypothesis 库生成float 的list。有一些重要的限制：

列表中的成分数量必须大于 1 且小于 15 最小值必须大于 0 最大值必须小于 1 所有成分的总和必须完全等于一 (1)

到目前为止，我能够满足前三个约束。

@given(
    strategies.lists(
        st.floats(min_value=0, max_value=1, exclude_min=True, exclude_max=True),
        min_size=2,
        max_size=15,
    )
)

我怎样才能满足第四个约束？

【参考方案1】：

我认为您不能直接添加约束，但您可以调整数据以使其满足条件，例如：

def normalize(float_list):
    s = sum(float_list)
    return [f / s for f in float_list]


@given(
    strategies.lists(
        st.floats(min_value=0, max_value=1, exclude_min=True,
                  exclude_max=True),
        min_size=2,
        max_size=15,
    ).map(normalize)
)
def test_sum(f):
    assert abs(sum(f) - 1) < 0.0000001

例如您自己对结果列表进行规范化，以便它通过条件。 请注意，这可能不会为您提供正好为 1 的数字（由于浮点数精度）。此外，假设可能会选择一些边缘情况（比如一些非常小的数字），这些情况在映射后可能不是边缘情况——这对您来说可能是也可能不是问题。