/
ExtendedStatsTest.php
72 lines (60 loc) · 2.04 KB
/
ExtendedStatsTest.php
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
<?php
namespace Elastica\Test\Aggregation;
use Elastica\Aggregation\ExtendedStats;
use Elastica\Document;
use Elastica\Index;
use Elastica\Query;
/**
* @internal
*/
class ExtendedStatsTest extends BaseAggregationTest
{
/**
* @group functional
*/
public function testExtendedStatsAggregation(): void
{
$agg = new ExtendedStats('stats');
$agg->setField('price');
$query = new Query();
$query->addAggregation($agg);
$results = $this->_getIndexForTest()->search($query)->getAggregation('stats');
$this->assertEquals(4, $results['count']);
$this->assertEquals(1, $results['min']);
$this->assertEquals(8, $results['max']);
$this->assertEquals((5 + 8 + 1 + 3) / 4.0, $results['avg']);
$this->assertEquals((5 + 8 + 1 + 3), $results['sum']);
$this->assertArrayHasKey('sum_of_squares', $results);
}
/**
* @group functional
*/
public function testExtendedStatsAggregationWithMissing(): void
{
$agg = new ExtendedStats('stats');
$agg->setField('price');
$agg->setMissing(10);
$query = new Query();
$query->addAggregation($agg);
$results = $this->_getIndexForTest()->search($query)->getAggregation('stats');
$this->assertEquals(5, $results['count']);
$this->assertEquals(1, $results['min']);
$this->assertEquals(10, $results['max']);
$this->assertEquals((5 + 8 + 1 + 3 + 10) / 5.0, $results['avg']);
$this->assertEquals((5 + 8 + 1 + 3 + 10), $results['sum']);
$this->assertArrayHasKey('sum_of_squares', $results);
}
protected function _getIndexForTest(): Index
{
$index = $this->_createIndex();
$index->addDocuments([
new Document(1, ['price' => 5]),
new Document(2, ['price' => 8]),
new Document(3, ['price' => 1]),
new Document(4, ['price' => 3]),
new Document(5, ['anything' => 'anything']),
]);
$index->refresh();
return $index;
}
}