Создайте два списка одинаковой длины в fixed_dictionaries, используя гипотезу - PullRequest
Новая
       6

Создайте два списка одинаковой длины в fixed_dictionaries, используя гипотезу

0 голосов
/ 21 сентября 2018

Я пытаюсь сгенерировать пример данных, используя стратегию 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) # ???

1 Ответ

0 голосов
/ 21 сентября 2018

Вот один из способов решить эту проблему: 

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