python - Pytest:根据假设生成总和为 1 的 float 列表

Question

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

1. 列表中的成分股数量必须大于 1 且小于 15
2. 最小值必须大于 0
3. 最大值必须小于 1
4. 所有成分之和必须恰好等于一 (1)

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

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

如何满足第四个约束？

Answer 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 的数字(由于 float 字精度)。此外，假设可能会选择一些边缘情况(例如一些非常小的数字)，这些情况在映射后可能不是边缘情况 - 这对您来说可能是问题，也可能不是问题。