ResInsight/ApplicationCode/UnitTests/RiaWeightedMean-Test.cpp

#include "gtest/gtest.h"

#include "RiaWeightedMeanCalculator.h"


//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(RiaWeightedMeanCalculator, BasicUsage)
{
    {
        RiaWeightedMeanCalculator<double> calc;

        EXPECT_DOUBLE_EQ(0.0, calc.aggregatedWeight());
        EXPECT_FALSE(calc.validAggregatedWeight());
    }

    {
        RiaWeightedMeanCalculator<double> calc;

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

        for (size_t i = 0; i< values.size(); i++)
        {
            calc.addValueAndWeight(values[i], weights[i]);
        }
        EXPECT_TRUE(calc.validAggregatedWeight());
        EXPECT_DOUBLE_EQ(3.0, calc.aggregatedWeight());
        EXPECT_DOUBLE_EQ(5.0, calc.weightedMean());
    }
}