diff options
Diffstat (limited to 'tests/test_asyncio/test_bloom.py')
| -rw-r--r-- | tests/test_asyncio/test_bloom.py | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/tests/test_asyncio/test_bloom.py b/tests/test_asyncio/test_bloom.py index 326924c..fd9e949 100644 --- a/tests/test_asyncio/test_bloom.py +++ b/tests/test_asyncio/test_bloom.py @@ -324,7 +324,7 @@ async def test_tdigest_reset(modclient: redis.Redis): # reset on empty histogram assert await modclient.tdigest().reset("tDigest") # insert data-points into sketch - assert await modclient.tdigest().add("tDigest", list(range(10)), [1] * 10) + assert await modclient.tdigest().add("tDigest", list(range(10))) assert await modclient.tdigest().reset("tDigest") # assert we have 0 unmerged nodes @@ -337,19 +337,19 @@ async def test_tdigest_merge(modclient: redis.Redis): assert await modclient.tdigest().create("to-tDigest", 10) assert await modclient.tdigest().create("from-tDigest", 10) # insert data-points into sketch - assert await modclient.tdigest().add("from-tDigest", [1.0] * 10, [1] * 10) - assert await modclient.tdigest().add("to-tDigest", [2.0] * 10, [10] * 10) + assert await modclient.tdigest().add("from-tDigest", [1.0] * 10) + assert await modclient.tdigest().add("to-tDigest", [2.0] * 10) # merge from-tdigest into to-tdigest assert await modclient.tdigest().merge("to-tDigest", 1, "from-tDigest") # we should now have 110 weight on to-histogram info = await modclient.tdigest().info("to-tDigest") total_weight_to = float(info.merged_weight) + float(info.unmerged_weight) - assert 110 == total_weight_to + assert 20.0 == total_weight_to # test override assert await modclient.tdigest().create("from-override", 10) assert await modclient.tdigest().create("from-override-2", 10) - assert await modclient.tdigest().add("from-override", [3.0] * 10, [10] * 10) - assert await modclient.tdigest().add("from-override-2", [4.0] * 10, [10] * 10) + assert await modclient.tdigest().add("from-override", [3.0] * 10) + assert await modclient.tdigest().add("from-override-2", [4.0] * 10) assert await modclient.tdigest().merge( "to-tDigest", 2, "from-override", "from-override-2", override=True ) @@ -362,7 +362,7 @@ async def test_tdigest_merge(modclient: redis.Redis): async def test_tdigest_min_and_max(modclient: redis.Redis): assert await modclient.tdigest().create("tDigest", 100) # insert data-points into sketch - assert await modclient.tdigest().add("tDigest", [1, 2, 3], [1] * 3) + assert await modclient.tdigest().add("tDigest", [1, 2, 3]) # min/max assert 3 == await modclient.tdigest().max("tDigest") assert 1 == await modclient.tdigest().min("tDigest") @@ -375,7 +375,7 @@ async def test_tdigest_quantile(modclient: redis.Redis): assert await modclient.tdigest().create("tDigest", 500) # insert data-points into sketch assert await modclient.tdigest().add( - "tDigest", list([x * 0.01 for x in range(1, 10000)]), [1] * 10000 + "tDigest", list([x * 0.01 for x in range(1, 10000)]) ) # assert min min/max have same result as quantile 0 and 1 assert ( @@ -392,7 +392,7 @@ async def test_tdigest_quantile(modclient: redis.Redis): # test multiple quantiles assert await modclient.tdigest().create("t-digest", 100) - assert await modclient.tdigest().add("t-digest", [1, 2, 3, 4, 5], [1] * 5) + assert await modclient.tdigest().add("t-digest", [1, 2, 3, 4, 5]) res = await modclient.tdigest().quantile("t-digest", 0.5, 0.8) assert [3.0, 5.0] == res @@ -402,7 +402,7 @@ async def test_tdigest_quantile(modclient: redis.Redis): async def test_tdigest_cdf(modclient: redis.Redis): assert await modclient.tdigest().create("tDigest", 100) # insert data-points into sketch - assert await modclient.tdigest().add("tDigest", list(range(1, 10)), [1] * 10) + assert await modclient.tdigest().add("tDigest", list(range(1, 10))) assert 0.1 == round((await modclient.tdigest().cdf("tDigest", 1.0))[0], 1) assert 0.9 == round((await modclient.tdigest().cdf("tDigest", 9.0))[0], 1) res = await modclient.tdigest().cdf("tDigest", 1.0, 9.0) @@ -415,7 +415,7 @@ async def test_tdigest_cdf(modclient: redis.Redis): async def test_tdigest_trimmed_mean(modclient: redis.Redis): assert await modclient.tdigest().create("tDigest", 100) # insert data-points into sketch - assert await modclient.tdigest().add("tDigest", list(range(1, 10)), [1] * 10) + assert await modclient.tdigest().add("tDigest", list(range(1, 10))) assert 5 == await modclient.tdigest().trimmed_mean("tDigest", 0.1, 0.9) assert 4.5 == await modclient.tdigest().trimmed_mean("tDigest", 0.4, 0.5) @@ -424,7 +424,7 @@ async def test_tdigest_trimmed_mean(modclient: redis.Redis): @pytest.mark.experimental async def test_tdigest_rank(modclient: redis.Redis): assert await modclient.tdigest().create("t-digest", 500) - assert await modclient.tdigest().add("t-digest", list(range(0, 20)), [1] * 20) + assert await modclient.tdigest().add("t-digest", list(range(0, 20))) assert -1 == (await modclient.tdigest().rank("t-digest", -1))[0] assert 1 == (await modclient.tdigest().rank("t-digest", 0))[0] assert 11 == (await modclient.tdigest().rank("t-digest", 10))[0] @@ -435,17 +435,17 @@ async def test_tdigest_rank(modclient: redis.Redis): @pytest.mark.experimental async def test_tdigest_revrank(modclient: redis.Redis): assert await modclient.tdigest().create("t-digest", 500) - assert await modclient.tdigest().add("t-digest", list(range(0, 20)), [1] * 20) + assert await modclient.tdigest().add("t-digest", list(range(0, 20))) assert -1 == (await modclient.tdigest().revrank("t-digest", 20))[0] - assert 20 == (await modclient.tdigest().revrank("t-digest", 0))[0] - assert [-1, 20, 10] == await modclient.tdigest().revrank("t-digest", 21, 0, 10) + assert 19 == (await modclient.tdigest().revrank("t-digest", 0))[0] + assert [-1, 19, 9] == await modclient.tdigest().revrank("t-digest", 21, 0, 10) @pytest.mark.redismod @pytest.mark.experimental async def test_tdigest_byrank(modclient: redis.Redis): assert await modclient.tdigest().create("t-digest", 500) - assert await modclient.tdigest().add("t-digest", list(range(1, 11)), [1] * 20) + assert await modclient.tdigest().add("t-digest", list(range(1, 11))) assert 1 == (await modclient.tdigest().byrank("t-digest", 0))[0] assert 10 == (await modclient.tdigest().byrank("t-digest", 9))[0] assert (await modclient.tdigest().byrank("t-digest", 100))[0] == inf @@ -457,9 +457,9 @@ async def test_tdigest_byrank(modclient: redis.Redis): @pytest.mark.experimental async def test_tdigest_byrevrank(modclient: redis.Redis): assert await modclient.tdigest().create("t-digest", 500) - assert await modclient.tdigest().add("t-digest", list(range(1, 11)), [1] * 20) + assert await modclient.tdigest().add("t-digest", list(range(1, 11))) assert 10 == (await modclient.tdigest().byrevrank("t-digest", 0))[0] - assert 2 == (await modclient.tdigest().byrevrank("t-digest", 9))[0] + assert 1 == (await modclient.tdigest().byrevrank("t-digest", 9))[0] assert (await modclient.tdigest().byrevrank("t-digest", 100))[0] == -inf with pytest.raises(redis.ResponseError): (await modclient.tdigest().byrevrank("t-digest", -1))[0] |