15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright 2013 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "remoting/base/running_average.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 77d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const int64 kTestValues[] = { 10, 20, 30, 10, 25, 16, 15 }; 11c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Average across a single element, i.e. just return the most recent. 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(RunningAverageTest, OneElementWindow) { 14a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) RunningAverage running_average(1); 15ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch EXPECT_EQ(0, running_average.Average()); 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < arraysize(kTestValues); ++i) { 1858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) running_average.Record(kTestValues[i]); 19bbcdd45c55eb7c4641ab97aef9889b0fc828e7d3Ben Murdoch EXPECT_EQ(static_cast<double>(kTestValues[i]), running_average.Average()); 203551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) } 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 23868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)// Average the two most recent elements. 24868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)TEST(RunningAverageTest, TwoElementWindow) { 25868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) RunningAverage running_average(2); 26f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) EXPECT_EQ(0, running_average.Average()); 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < arraysize(kTestValues); ++i) { 290f1bc08d4cfcc34181b0b5cbf065c40f687bf740Torne (Richard Coles) running_average.Record(kTestValues[i]); 30f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 31ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch double expected = kTestValues[i]; 32a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch if (i > 0) 33a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch expected = (expected + kTestValues[i-1]) / 2; 347dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 35eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch EXPECT_EQ(expected, running_average.Average()); 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// Average across all the elements if the window size exceeds the element count. 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(RunningAverageTest, LongWindow) { 417dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch RunningAverage running_average(arraysize(kTestValues) + 1); 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(0, running_average.Average()); 43a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < arraysize(kTestValues); ++i) { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) running_average.Record(kTestValues[i]); 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) double expected = 0.0; 48c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) for (size_t j = 0; j <= i; ++j) 497dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch expected += kTestValues[j]; 50c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) expected /= i + 1; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ(expected, running_average.Average()); 535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)