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

Question

对于单元测试，我想生成一个 list的 float使用 hypothesis图书馆。有一些重要的限制:

列表中的成分股数量必须大于 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 的数字(由于浮点数精度)。此外，假设可能会选择一些边缘情况(例如一些非常小的数字)，这些情况在映射后可能不是边缘情况——这对您来说可能是也可能不是问题。