python - 使用假设在 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))