#3210 Helper class used to compute weighted average

This commit is contained in:
Magne Sjaastad
2018-08-15 08:38:54 +02:00
parent f6e6cacc36
commit e195eed246
5 changed files with 131 additions and 0 deletions

View File

@@ -47,6 +47,7 @@ ${CMAKE_CURRENT_LIST_DIR}/RiaTimeHistoryCurveTools-Test.cpp
${CMAKE_CURRENT_LIST_DIR}/SolveSpaceSolver-Test.cpp
${CMAKE_CURRENT_LIST_DIR}/RiaPolyArcLineSampler-Test.cpp
${CMAKE_CURRENT_LIST_DIR}/RifEclipseDataTableFormatter-Test.cpp
${CMAKE_CURRENT_LIST_DIR}/RiaWeightedAverage-Test.cpp
)
list(APPEND CODE_HEADER_FILES

View File

@@ -0,0 +1,33 @@
#include "gtest/gtest.h"
#include "RiaWeightedAverageCalculator.h"
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
TEST(RiaWeightedAverageCalculator, BasicUsage)
{
{
RiaWeightedAverageCalculator calc;
EXPECT_DOUBLE_EQ(0.0, calc.aggregatedWeight());
EXPECT_DOUBLE_EQ(0.0, calc.weightedAverage());
}
{
RiaWeightedAverageCalculator 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_DOUBLE_EQ(3.0, calc.aggregatedWeight());
EXPECT_DOUBLE_EQ(5.0, calc.weightedAverage());
}
}