1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
from .base import BaseTest, st, commands, keys, fields, common_commands
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))
)
# TODO:
# - find a way to test srandmember, spop which are random
# - sscan
create_command_strategy = commands(st.just("sadd"), keys, st.lists(fields, min_size=1))
command_strategy = set_commands | common_commands
|
