ResInsight/ApplicationCode/UnitTests/RiaWeightedGeometricMeanCalculator-Test.cpp

#include "gtest/gtest.h"

#include "RiaWeightedGeometricMeanCalculator.h"


//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(RiaWeightedGeometricMeanCalculator, BasicUsage)
{
    {
        RiaWeightedGeometricMeanCalculator calc;

        EXPECT_DOUBLE_EQ(0.0, calc.aggregatedWeight());
        EXPECT_DOUBLE_EQ(0.0, calc.weightedMean());
    }


    {
        RiaWeightedGeometricMeanCalculator calc;

        std::vector<double> values {30.0, 60.0};
        std::vector<double> weights {1.5, 3.5};

        for (size_t i = 0; i< values.size(); i++)
        {
            calc.addValueAndWeight(values[i], weights[i]);
        }

        double expectedValue = std::pow(
            std::pow(30.0, 1.5) * std::pow(60.0, 3.5),
            1 / (1.5 + 3.5)
        );

        EXPECT_DOUBLE_EQ(5.0, calc.aggregatedWeight());
        EXPECT_DOUBLE_EQ(expectedValue, calc.weightedMean());
    }
}