summaryrefslogtreecommitdiffstats
path: root/eval/src/tests/eval/aggr/aggr_test.cpp
blob: d81d6bb0ff34438bbbccc526379b68c4c579a666 (plain) (blame)
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
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/vespalib/testkit/test_kit.h>
#include <vespa/vespalib/util/stash.h>
#include <vespa/eval/eval/aggr.h>

using vespalib::Stash;
using namespace vespalib::eval;

TEST("require that AVG aggregator works as expected") {
    Stash stash;
    Aggregator &aggr = Aggregator::create(Aggr::AVG, stash);
    EXPECT_EQUAL(aggr.result(), 0.0);
    aggr.first(10.0),  EXPECT_EQUAL(aggr.result(), 10.0);
    aggr.next(20.0),   EXPECT_EQUAL(aggr.result(), 15.0);
    aggr.next(30.0),   EXPECT_EQUAL(aggr.result(), 20.0);
    aggr.first(100.0), EXPECT_EQUAL(aggr.result(), 100.0);
    aggr.next(200.0),  EXPECT_EQUAL(aggr.result(), 150.0);
}

TEST("require that COUNT aggregator works as expected") {
    Stash stash;
    Aggregator &aggr = Aggregator::create(Aggr::COUNT, stash);
    EXPECT_EQUAL(aggr.result(), 0.0);
    aggr.first(10.0),  EXPECT_EQUAL(aggr.result(), 1.0);
    aggr.next(20.0),   EXPECT_EQUAL(aggr.result(), 2.0);
    aggr.next(30.0),   EXPECT_EQUAL(aggr.result(), 3.0);
    aggr.first(100.0), EXPECT_EQUAL(aggr.result(), 1.0);
    aggr.next(200.0),  EXPECT_EQUAL(aggr.result(), 2.0);
}

TEST("require that PROD aggregator works as expected") {
    Stash stash;
    Aggregator &aggr = Aggregator::create(Aggr::PROD, stash);
    EXPECT_EQUAL(aggr.result(), 0.0);
    aggr.first(10.0),  EXPECT_EQUAL(aggr.result(), 10.0);
    aggr.next(20.0),   EXPECT_EQUAL(aggr.result(), 200.0);
    aggr.next(30.0),   EXPECT_EQUAL(aggr.result(), 6000.0);
    aggr.first(100.0), EXPECT_EQUAL(aggr.result(), 100.0);
    aggr.next(200.0),  EXPECT_EQUAL(aggr.result(), 20000.0);
}

TEST("require that SUM aggregator works as expected") {
    Stash stash;
    Aggregator &aggr = Aggregator::create(Aggr::SUM, stash);
    EXPECT_EQUAL(aggr.result(), 0.0);
    aggr.first(10.0),  EXPECT_EQUAL(aggr.result(), 10.0);
    aggr.next(20.0),   EXPECT_EQUAL(aggr.result(), 30.0);
    aggr.next(30.0),   EXPECT_EQUAL(aggr.result(), 60.0);
    aggr.first(100.0), EXPECT_EQUAL(aggr.result(), 100.0);
    aggr.next(200.0),  EXPECT_EQUAL(aggr.result(), 300.0);
}

TEST("require that MAX aggregator works as expected") {
    Stash stash;
    Aggregator &aggr = Aggregator::create(Aggr::MAX, stash);
    EXPECT_EQUAL(aggr.result(), 0.0);
    aggr.first(10.0),  EXPECT_EQUAL(aggr.result(), 10.0);
    aggr.next(20.0),   EXPECT_EQUAL(aggr.result(), 20.0);
    aggr.next(30.0),   EXPECT_EQUAL(aggr.result(), 30.0);
    aggr.first(100.0), EXPECT_EQUAL(aggr.result(), 100.0);
    aggr.next(200.0),  EXPECT_EQUAL(aggr.result(), 200.0);
}

TEST("require that MIN aggregator works as expected") {
    Stash stash;
    Aggregator &aggr = Aggregator::create(Aggr::MIN, stash);
    EXPECT_EQUAL(aggr.result(), 0.0);
    aggr.first(10.0),  EXPECT_EQUAL(aggr.result(), 10.0);
    aggr.next(20.0),   EXPECT_EQUAL(aggr.result(), 10.0);
    aggr.next(30.0),   EXPECT_EQUAL(aggr.result(), 10.0);
    aggr.first(100.0), EXPECT_EQUAL(aggr.result(), 100.0);
    aggr.next(200.0),  EXPECT_EQUAL(aggr.result(), 100.0);
}

TEST_MAIN() { TEST_RUN_ALL(); }