|
| 1 | +// EXAMPLE: tdigest_tutorial |
| 2 | +<?php |
| 3 | + |
| 4 | +require 'vendor/autoload.php'; |
| 5 | + |
| 6 | +use Predis\Client as PredisClient; |
| 7 | + |
| 8 | +class DtTdigestTest |
| 9 | +// REMOVE_START |
| 10 | +extends PredisTestCase |
| 11 | +// REMOVE_END |
| 12 | +{ |
| 13 | + public function testDtTdigest() { |
| 14 | + $r = new PredisClient([ |
| 15 | + 'scheme' => 'tcp', |
| 16 | + 'host' => '127.0.0.1', |
| 17 | + 'port' => 6379, |
| 18 | + 'password' => '', |
| 19 | + 'database' => 0, |
| 20 | + ]); |
| 21 | + // REMOVE_START |
| 22 | + $r->flushall(); |
| 23 | + // REMOVE_END |
| 24 | + |
| 25 | + // STEP_START tdig_start |
| 26 | + $res1 = $r->tdigestcreate('bikes:sales', 100); |
| 27 | + echo $res1 . PHP_EOL; |
| 28 | + // >>> OK |
| 29 | + |
| 30 | + $res2 = $r->tdigestadd('bikes:sales', 21); |
| 31 | + echo $res2 . PHP_EOL; |
| 32 | + // >>> OK |
| 33 | + |
| 34 | + $res3 = $r->tdigestadd('bikes:sales', 150, 95, 75, 34); |
| 35 | + echo $res3 . PHP_EOL; |
| 36 | + // >>> OK |
| 37 | + // STEP_END |
| 38 | + // REMOVE_START |
| 39 | + $this->assertEquals('OK', $res1); |
| 40 | + $this->assertEquals('OK', $res2); |
| 41 | + $this->assertEquals('OK', $res3); |
| 42 | + // REMOVE_END |
| 43 | + |
| 44 | + // STEP_START tdig_cdf |
| 45 | + $res4 = $r->tdigestcreate('racer_ages'); |
| 46 | + echo $res4 . PHP_EOL; |
| 47 | + // >>> OK |
| 48 | + |
| 49 | + $res5 = $r->tdigestadd( |
| 50 | + 'racer_ages', |
| 51 | + 45.88, |
| 52 | + 44.2, |
| 53 | + 58.03, |
| 54 | + 19.76, |
| 55 | + 39.84, |
| 56 | + 69.28, |
| 57 | + 50.97, |
| 58 | + 25.41, |
| 59 | + 19.27, |
| 60 | + 85.71, |
| 61 | + 42.63 |
| 62 | + ); |
| 63 | + echo $res5 . PHP_EOL; |
| 64 | + // >>> OK |
| 65 | + |
| 66 | + $res6 = $r->tdigestrank('racer_ages', 50); |
| 67 | + echo json_encode($res6) . PHP_EOL; |
| 68 | + // >>> [7] |
| 69 | + |
| 70 | + $res7 = $r->tdigestrank('racer_ages', 50, 40); |
| 71 | + echo json_encode($res7) . PHP_EOL; |
| 72 | + // >>> [7,4] |
| 73 | + // STEP_END |
| 74 | + // REMOVE_START |
| 75 | + $this->assertEquals('OK', $res4); |
| 76 | + $this->assertEquals('OK', $res5); |
| 77 | + $this->assertEquals([7], $res6); |
| 78 | + $this->assertEquals([7, 4], $res7); |
| 79 | + // REMOVE_END |
| 80 | + |
| 81 | + // STEP_START tdig_quant |
| 82 | + $res8 = $r->tdigestquantile('racer_ages', 0.5); |
| 83 | + echo json_encode($res8) . PHP_EOL; |
| 84 | + // >>> ["44.2"] |
| 85 | + |
| 86 | + $res9 = $r->tdigestbyrank('racer_ages', 4); |
| 87 | + echo json_encode($res9) . PHP_EOL; |
| 88 | + // >>> ["42.63"] |
| 89 | + // STEP_END |
| 90 | + // REMOVE_START |
| 91 | + $this->assertEquals(['44.2'], $res8); |
| 92 | + $this->assertEquals(['42.63'], $res9); |
| 93 | + // REMOVE_END |
| 94 | + |
| 95 | + // STEP_START tdig_min |
| 96 | + $res10 = $r->tdigestmin('racer_ages'); |
| 97 | + echo $res10 . PHP_EOL; |
| 98 | + // >>> 19.27 |
| 99 | + |
| 100 | + $res11 = $r->tdigestmax('racer_ages'); |
| 101 | + echo $res11 . PHP_EOL; |
| 102 | + // >>> 85.71 |
| 103 | + // STEP_END |
| 104 | + // REMOVE_START |
| 105 | + $this->assertEquals('19.27', $res10); |
| 106 | + $this->assertEquals('85.71', $res11); |
| 107 | + // REMOVE_END |
| 108 | + |
| 109 | + // STEP_START tdig_reset |
| 110 | + $res12 = $r->tdigestreset('racer_ages'); |
| 111 | + echo $res12 . PHP_EOL; |
| 112 | + // >>> OK |
| 113 | + // STEP_END |
| 114 | + // REMOVE_START |
| 115 | + $this->assertEquals('OK', $res12); |
| 116 | + // REMOVE_END |
| 117 | + } |
| 118 | +} |
| 119 | + |
0 commit comments