Pytest：生成总计为 1 的浮点数列表与假设

Question

对于单元测试，我想使用 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,
    )
)

如何满足第四个约束？

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 的数字（由于浮点数精度）。此外，假设可能会选择一些边缘情况（比如一些非常小的数字），这些情况在映射后可能不是边缘情况——这对你来说可能是问题，也可能不是问题。

Pytest：生成总计为 1 的浮点数列表与假设

Pytest: generating list of floats summing up to 1 with hypothesis

python

pytest

hypothesis-test