python - 使用 Hypothesis 在 fixed_dictionaries 中生成两个长度相等的列表

Question

我正在尝试使用fixed_dictionaries其中两个键具有列表作为必须具有相同长度的值的策略来生成示例数据，例如：

{'ids': [1, 2, 3],
 'words': ['foo', 'bar', 'baz']}

如何强制执行此约束？我想我也许可以将一个定义为另一个的组合，但我不知道该怎么做。就像是：

import hypothesis.strategies as st

ids = st.lists(elements=st.integers())

@st.composite
def words(draw, elements=st.text()):
    draw(sample_ids) # ???

Answer 1

这是解决此问题的一种方法：

@composite
def my_dicts(draw):
    size = draw(st.integers(min_value=0))
    ids = st.lists(min_size=size, max_size=size)
    words = st.lists(min_size=size, max_size=size)
    dicts = st.fixed_dictionaries({'ids': ids, 'words': words})

    return draw(st.lists(dicts))