ResInsight/ApplicationLibCode/UnitTests/RiaMedianCalculator-Test.cpp at dev · OPM/ResInsight · GitHub

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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#include "gtest/gtest.h"

#include "RiaMedianCalculator.h"

#include <algorithm>
#include <random>
#include <vector>

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
TEST( RiaMedianCalculator, EmptyDataSet )
{
    RiaMedianCalculator<double> calc;

    EXPECT_FALSE( calc.valid() );
}

TEST( RiaMedianCalculator, SingleValue )
{
    RiaMedianCalculator<double> calc;
    calc.add( 4.0 );
    EXPECT_DOUBLE_EQ( 4.0, calc.median() );
}

TEST( RiaMedianCalculator, TwoValues )
{
    RiaMedianCalculator<double> calc;

    std::vector<double> values{ 3.0, 6.0 };

    for ( size_t i = 0; i < values.size(); i++ )
    {
        calc.add( values[i] );
    }
    EXPECT_TRUE( calc.valid() );
    EXPECT_DOUBLE_EQ( 4.5, calc.median() );
}

TEST( RiaMedianCalculator, ThreeValues )
{
    RiaMedianCalculator<double> calc;

    std::vector<double> values{ 3.0, 6.0, 2.0 };

    for ( size_t i = 0; i < values.size(); i++ )
    {
        calc.add( values[i] );
    }
    EXPECT_TRUE( calc.valid() );
    EXPECT_DOUBLE_EQ( 3.0, calc.median() );
}

TEST( RiaMedianCalculator, SameValues )
{
    RiaMedianCalculator<double> calc;
    const double                value = 113.0;
    for ( size_t i = 0; i < 1000; ++i )
    {
        calc.add( value );
    }
    EXPECT_TRUE( calc.valid() );
    EXPECT_DOUBLE_EQ( value, calc.median() );
}

TEST( RiaMedianCalculator, OrderedRangeOdd )
{
    RiaMedianCalculator<double> calc;
    for ( size_t i = 0; i < 101; ++i )
    {
        calc.add( (double)i );
    }
    EXPECT_TRUE( calc.valid() );
    EXPECT_DOUBLE_EQ( 50.0, calc.median() );
}

TEST( RiaMedianCalculator, OrderedRangeEven )
{
    RiaMedianCalculator<double> calc;
    for ( size_t i = 0; i < 100; ++i )
    {
        calc.add( (double)i );
    }
    EXPECT_TRUE( calc.valid() );
    EXPECT_DOUBLE_EQ( 49.5, calc.median() );
}

TEST( RiaMedianCalculator, ShuffledRangeOdd )
{
    RiaMedianCalculator<double> calc;

    std::vector<int> values;
    for ( int i = 0; i < 101; ++i )
    {
        values.push_back( i );
    }
    std::random_device rd;
    std::mt19937       g( rd() );
    std::shuffle( values.begin(), values.end(), g );
    for ( double value : values )
    {
        calc.add( value );
    }
    EXPECT_TRUE( calc.valid() );
    EXPECT_DOUBLE_EQ( 50.0, calc.median() );
}

TEST( RiaMedianCalculator, ShuffledRangeEven )
{
    RiaMedianCalculator<double> calc;

    std::vector<int> values;
    for ( int i = 0; i < 200; ++i )
    {
        values.push_back( i );
    }
    std::random_device rd;
    std::mt19937       g( rd() );
    std::shuffle( values.begin(), values.end(), g );
    for ( double value : values )
    {
        calc.add( value );
    }

    EXPECT_TRUE( calc.valid() );
    EXPECT_DOUBLE_EQ( 99.5, calc.median() );
}