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))

使用假设在fixed_dictionaries中生成两个等长列表

1 个答案: