@@ -324,7 +324,7 @@ async def test_tdigest_reset(modclient: redis.Redis):
324
324
# reset on empty histogram
325
325
assert await modclient .tdigest ().reset ("tDigest" )
326
326
# insert data-points into sketch
327
- assert await modclient .tdigest ().add ("tDigest" , list (range (10 )), [ 1 ] * 10 )
327
+ assert await modclient .tdigest ().add ("tDigest" , list (range (10 )))
328
328
329
329
assert await modclient .tdigest ().reset ("tDigest" )
330
330
# assert we have 0 unmerged nodes
@@ -337,19 +337,19 @@ async def test_tdigest_merge(modclient: redis.Redis):
337
337
assert await modclient .tdigest ().create ("to-tDigest" , 10 )
338
338
assert await modclient .tdigest ().create ("from-tDigest" , 10 )
339
339
# insert data-points into sketch
340
- assert await modclient .tdigest ().add ("from-tDigest" , [1.0 ] * 10 , [ 1 ] * 10 )
341
- assert await modclient .tdigest ().add ("to-tDigest" , [2.0 ] * 10 , [ 10 ] * 10 )
340
+ assert await modclient .tdigest ().add ("from-tDigest" , [1.0 ] * 10 )
341
+ assert await modclient .tdigest ().add ("to-tDigest" , [2.0 ] * 10 )
342
342
# merge from-tdigest into to-tdigest
343
343
assert await modclient .tdigest ().merge ("to-tDigest" , 1 , "from-tDigest" )
344
344
# we should now have 110 weight on to-histogram
345
345
info = await modclient .tdigest ().info ("to-tDigest" )
346
346
total_weight_to = float (info .merged_weight ) + float (info .unmerged_weight )
347
- assert 110 == total_weight_to
347
+ assert 20.0 == total_weight_to
348
348
# test override
349
349
assert await modclient .tdigest ().create ("from-override" , 10 )
350
350
assert await modclient .tdigest ().create ("from-override-2" , 10 )
351
- assert await modclient .tdigest ().add ("from-override" , [3.0 ] * 10 , [ 10 ] * 10 )
352
- assert await modclient .tdigest ().add ("from-override-2" , [4.0 ] * 10 , [ 10 ] * 10 )
351
+ assert await modclient .tdigest ().add ("from-override" , [3.0 ] * 10 )
352
+ assert await modclient .tdigest ().add ("from-override-2" , [4.0 ] * 10 )
353
353
assert await modclient .tdigest ().merge (
354
354
"to-tDigest" , 2 , "from-override" , "from-override-2" , override = True
355
355
)
@@ -362,7 +362,7 @@ async def test_tdigest_merge(modclient: redis.Redis):
362
362
async def test_tdigest_min_and_max (modclient : redis .Redis ):
363
363
assert await modclient .tdigest ().create ("tDigest" , 100 )
364
364
# insert data-points into sketch
365
- assert await modclient .tdigest ().add ("tDigest" , [1 , 2 , 3 ], [ 1 ] * 3 )
365
+ assert await modclient .tdigest ().add ("tDigest" , [1 , 2 , 3 ])
366
366
# min/max
367
367
assert 3 == await modclient .tdigest ().max ("tDigest" )
368
368
assert 1 == await modclient .tdigest ().min ("tDigest" )
@@ -375,7 +375,7 @@ async def test_tdigest_quantile(modclient: redis.Redis):
375
375
assert await modclient .tdigest ().create ("tDigest" , 500 )
376
376
# insert data-points into sketch
377
377
assert await modclient .tdigest ().add (
378
- "tDigest" , list ([x * 0.01 for x in range (1 , 10000 )]), [ 1 ] * 10000
378
+ "tDigest" , list ([x * 0.01 for x in range (1 , 10000 )])
379
379
)
380
380
# assert min min/max have same result as quantile 0 and 1
381
381
assert (
@@ -392,7 +392,7 @@ async def test_tdigest_quantile(modclient: redis.Redis):
392
392
393
393
# test multiple quantiles
394
394
assert await modclient .tdigest ().create ("t-digest" , 100 )
395
- assert await modclient .tdigest ().add ("t-digest" , [1 , 2 , 3 , 4 , 5 ], [ 1 ] * 5 )
395
+ assert await modclient .tdigest ().add ("t-digest" , [1 , 2 , 3 , 4 , 5 ])
396
396
res = await modclient .tdigest ().quantile ("t-digest" , 0.5 , 0.8 )
397
397
assert [3.0 , 5.0 ] == res
398
398
@@ -402,7 +402,7 @@ async def test_tdigest_quantile(modclient: redis.Redis):
402
402
async def test_tdigest_cdf (modclient : redis .Redis ):
403
403
assert await modclient .tdigest ().create ("tDigest" , 100 )
404
404
# insert data-points into sketch
405
- assert await modclient .tdigest ().add ("tDigest" , list (range (1 , 10 )), [ 1 ] * 10 )
405
+ assert await modclient .tdigest ().add ("tDigest" , list (range (1 , 10 )))
406
406
assert 0.1 == round ((await modclient .tdigest ().cdf ("tDigest" , 1.0 ))[0 ], 1 )
407
407
assert 0.9 == round ((await modclient .tdigest ().cdf ("tDigest" , 9.0 ))[0 ], 1 )
408
408
res = await modclient .tdigest ().cdf ("tDigest" , 1.0 , 9.0 )
@@ -415,7 +415,7 @@ async def test_tdigest_cdf(modclient: redis.Redis):
415
415
async def test_tdigest_trimmed_mean (modclient : redis .Redis ):
416
416
assert await modclient .tdigest ().create ("tDigest" , 100 )
417
417
# insert data-points into sketch
418
- assert await modclient .tdigest ().add ("tDigest" , list (range (1 , 10 )), [ 1 ] * 10 )
418
+ assert await modclient .tdigest ().add ("tDigest" , list (range (1 , 10 )))
419
419
assert 5 == await modclient .tdigest ().trimmed_mean ("tDigest" , 0.1 , 0.9 )
420
420
assert 4.5 == await modclient .tdigest ().trimmed_mean ("tDigest" , 0.4 , 0.5 )
421
421
@@ -424,7 +424,7 @@ async def test_tdigest_trimmed_mean(modclient: redis.Redis):
424
424
@pytest .mark .experimental
425
425
async def test_tdigest_rank (modclient : redis .Redis ):
426
426
assert await modclient .tdigest ().create ("t-digest" , 500 )
427
- assert await modclient .tdigest ().add ("t-digest" , list (range (0 , 20 )), [ 1 ] * 20 )
427
+ assert await modclient .tdigest ().add ("t-digest" , list (range (0 , 20 )))
428
428
assert - 1 == (await modclient .tdigest ().rank ("t-digest" , - 1 ))[0 ]
429
429
assert 1 == (await modclient .tdigest ().rank ("t-digest" , 0 ))[0 ]
430
430
assert 11 == (await modclient .tdigest ().rank ("t-digest" , 10 ))[0 ]
@@ -435,17 +435,17 @@ async def test_tdigest_rank(modclient: redis.Redis):
435
435
@pytest .mark .experimental
436
436
async def test_tdigest_revrank (modclient : redis .Redis ):
437
437
assert await modclient .tdigest ().create ("t-digest" , 500 )
438
- assert await modclient .tdigest ().add ("t-digest" , list (range (0 , 20 )), [ 1 ] * 20 )
438
+ assert await modclient .tdigest ().add ("t-digest" , list (range (0 , 20 )))
439
439
assert - 1 == (await modclient .tdigest ().revrank ("t-digest" , 20 ))[0 ]
440
- assert 20 == (await modclient .tdigest ().revrank ("t-digest" , 0 ))[0 ]
441
- assert [- 1 , 20 , 10 ] == await modclient .tdigest ().revrank ("t-digest" , 21 , 0 , 10 )
440
+ assert 19 == (await modclient .tdigest ().revrank ("t-digest" , 0 ))[0 ]
441
+ assert [- 1 , 19 , 9 ] == await modclient .tdigest ().revrank ("t-digest" , 21 , 0 , 10 )
442
442
443
443
444
444
@pytest .mark .redismod
445
445
@pytest .mark .experimental
446
446
async def test_tdigest_byrank (modclient : redis .Redis ):
447
447
assert await modclient .tdigest ().create ("t-digest" , 500 )
448
- assert await modclient .tdigest ().add ("t-digest" , list (range (1 , 11 )), [ 1 ] * 20 )
448
+ assert await modclient .tdigest ().add ("t-digest" , list (range (1 , 11 )))
449
449
assert 1 == (await modclient .tdigest ().byrank ("t-digest" , 0 ))[0 ]
450
450
assert 10 == (await modclient .tdigest ().byrank ("t-digest" , 9 ))[0 ]
451
451
assert (await modclient .tdigest ().byrank ("t-digest" , 100 ))[0 ] == inf
@@ -457,9 +457,9 @@ async def test_tdigest_byrank(modclient: redis.Redis):
457
457
@pytest .mark .experimental
458
458
async def test_tdigest_byrevrank (modclient : redis .Redis ):
459
459
assert await modclient .tdigest ().create ("t-digest" , 500 )
460
- assert await modclient .tdigest ().add ("t-digest" , list (range (1 , 11 )), [ 1 ] * 20 )
460
+ assert await modclient .tdigest ().add ("t-digest" , list (range (1 , 11 )))
461
461
assert 10 == (await modclient .tdigest ().byrevrank ("t-digest" , 0 ))[0 ]
462
- assert 2 == (await modclient .tdigest ().byrevrank ("t-digest" , 9 ))[0 ]
462
+ assert 1 == (await modclient .tdigest ().byrevrank ("t-digest" , 9 ))[0 ]
463
463
assert (await modclient .tdigest ().byrevrank ("t-digest" , 100 ))[0 ] == - inf
464
464
with pytest .raises (redis .ResponseError ):
465
465
(await modclient .tdigest ().byrevrank ("t-digest" , - 1 ))[0 ]
0 commit comments