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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
import pytest
from _testutils import redis_version
pytestmark = redis_version(
2, 8, 9, reason='HyperLogLog works only with redis>=2.8.9')
async def test_pfcount(redis):
key = 'hll_pfcount'
other_key = 'some-other-hll'
# add initial data, cardinality changed so command returns 1
is_changed = await redis.pfadd(key, 'foo', 'bar', 'zap')
assert is_changed == 1
# add more data, cardinality not changed so command returns 0
is_changed = await redis.pfadd(key, 'zap', 'zap', 'zap')
assert is_changed == 0
# add event more data, cardinality not changed so command returns 0
is_changed = await redis.pfadd(key, 'foo', 'bar')
assert is_changed == 0
# check cardinality of one key
cardinality = await redis.pfcount(key)
assert cardinality == 3
# create new key (variable) for cardinality estimation
is_changed = await redis.pfadd(other_key, 1, 2, 3)
assert is_changed == 1
# check cardinality of multiple keys
cardinality = await redis.pfcount(key, other_key)
assert cardinality == 6
with pytest.raises(TypeError):
await redis.pfcount(None)
with pytest.raises(TypeError):
await redis.pfcount(key, None)
with pytest.raises(TypeError):
await redis.pfcount(key, key, None)
async def test_pfadd(redis):
key = 'hll_pfadd'
values = ['a', 's', 'y', 'n', 'c', 'i', 'o']
# add initial data, cardinality changed so command returns 1
is_changed = await redis.pfadd(key, *values)
assert is_changed == 1
# add event more data, cardinality not changed so command returns 0
is_changed = await redis.pfadd(key, 'i', 'o')
assert is_changed == 0
async def test_pfadd_wrong_input(redis):
with pytest.raises(TypeError):
await redis.pfadd(None, 'value')
async def test_pfmerge(redis):
key = 'hll_asyncio'
key_other = 'hll_aioredis'
key_dest = 'hll_aio'
values = ['a', 's', 'y', 'n', 'c', 'i', 'o']
values_other = ['a', 'i', 'o', 'r', 'e', 'd', 'i', 's']
data_set = set(values + values_other)
cardinality_merged = len(data_set)
# add initial data, cardinality changed so command returns 1
await redis.pfadd(key, *values)
await redis.pfadd(key_other, *values_other)
# check cardinality of one key
cardinality = await redis.pfcount(key)
assert cardinality == len(set(values_other))
cardinality_other = await redis.pfcount(key_other)
assert cardinality_other == len(set(values_other))
await redis.pfmerge(key_dest, key, key_other)
cardinality_dest = await redis.pfcount(key_dest)
assert cardinality_dest == cardinality_merged
with pytest.raises(TypeError):
await redis.pfmerge(None, key)
with pytest.raises(TypeError):
await redis.pfmerge(key_dest, None)
with pytest.raises(TypeError):
await redis.pfmerge(key_dest, key, None)
async def test_pfmerge_wrong_input(redis):
with pytest.raises(TypeError):
await redis.pfmerge(None, 'value')
|
