dragonfly/tests/fakeredis/test/test_hypothesis/test_set.py
Daniel M c06e154d43
Fix test hypothesis (#4927)
* fix:test-hypothesis

---------

Co-authored-by: Roman Gershman <roman@dragonflydb.io>
2025-04-23 18:11:37 +03:00

30 lines
896 B
Python

import hypothesis.strategies as st
from .base import (
BaseTest,
commands,
keys,
common_commands,
fields,
)
class TestSet(BaseTest):
set_commands = (
commands(st.just("sadd"), keys, st.lists(fields))
| commands(st.just("scard"), keys)
| commands(st.sampled_from(["sdiff", "sinter", "sunion"]), st.lists(keys))
| commands(
st.sampled_from(["sdiffstore", "sinterstore", "sunionstore"]),
keys,
st.lists(keys),
)
| commands(st.just("sismember"), keys, fields)
| commands(st.just("smembers"), keys)
| commands(st.just("smove"), keys, keys, fields)
| commands(st.just("srem"), keys, st.lists(fields))
)
create_command_strategy = commands(
st.just("sadd"), keys, st.lists(fields, min_size=1)
)
command_strategy = set_commands | common_commands