moving_average_unittest.cc 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120
  1. // Copyright 2020 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "base/power_monitor/moving_average.h"
  5. #include "testing/gtest/include/gtest/gtest.h"
  6. namespace base {
  7. namespace test {
  8. // Ported from third_party/webrtc/rtc_base/numerics/moving_average_unittest.cc.
  9. TEST(MovingAverageTest, EmptyAverage) {
  10. MovingAverage moving_average(1);
  11. EXPECT_EQ(0u, moving_average.Size());
  12. EXPECT_EQ(0, moving_average.GetAverageRoundedDown());
  13. }
  14. TEST(MovingAverageTest, OneElement) {
  15. MovingAverage moving_average(1);
  16. moving_average.AddSample(3);
  17. EXPECT_EQ(1u, moving_average.Size());
  18. EXPECT_EQ(3, moving_average.GetAverageRoundedDown());
  19. }
  20. // Verify that Size() increases monotonically when samples are added up to the
  21. // window size. At that point the filter is full and shall return the window
  22. // size as Size() until Reset() is called.
  23. TEST(MovingAverageTest, Size) {
  24. constexpr uint8_t kWindowSize = 3;
  25. MovingAverage moving_average(kWindowSize);
  26. EXPECT_EQ(0u, moving_average.Size());
  27. moving_average.AddSample(1);
  28. EXPECT_EQ(1u, moving_average.Size());
  29. moving_average.AddSample(2);
  30. EXPECT_EQ(2u, moving_average.Size());
  31. moving_average.AddSample(3);
  32. // Three samples have beend added and the filter is full (all elements in the
  33. // buffer have been given a valid value).
  34. EXPECT_EQ(kWindowSize, moving_average.Size());
  35. // Adding a fourth sample will shift out the first sample (1) and the filter
  36. // should now contain [4,2,3] => average is 9 / 3 = 3.
  37. moving_average.AddSample(4);
  38. EXPECT_EQ(kWindowSize, moving_average.Size());
  39. EXPECT_EQ(3, moving_average.GetAverageRoundedToClosest());
  40. moving_average.Reset();
  41. EXPECT_EQ(0u, moving_average.Size());
  42. EXPECT_EQ(0, moving_average.GetAverageRoundedToClosest());
  43. }
  44. TEST(MovingAverageTest, GetAverage) {
  45. MovingAverage moving_average(255);
  46. moving_average.AddSample(1);
  47. moving_average.AddSample(1);
  48. moving_average.AddSample(3);
  49. moving_average.AddSample(3);
  50. EXPECT_EQ(moving_average.GetAverageRoundedDown(), 2);
  51. EXPECT_EQ(moving_average.GetAverageRoundedToClosest(), 2);
  52. }
  53. TEST(MovingAverageTest, GetAverageRoundedDownRounds) {
  54. MovingAverage moving_average(255);
  55. moving_average.AddSample(1);
  56. moving_average.AddSample(2);
  57. moving_average.AddSample(2);
  58. moving_average.AddSample(2);
  59. EXPECT_EQ(moving_average.GetAverageRoundedDown(), 1);
  60. }
  61. TEST(MovingAverageTest, GetAverageRoundedToClosestRounds) {
  62. MovingAverage moving_average(255);
  63. moving_average.AddSample(1);
  64. moving_average.AddSample(2);
  65. moving_average.AddSample(2);
  66. moving_average.AddSample(2);
  67. EXPECT_EQ(moving_average.GetAverageRoundedToClosest(), 2);
  68. }
  69. TEST(MovingAverageTest, Reset) {
  70. MovingAverage moving_average(5);
  71. moving_average.AddSample(1);
  72. EXPECT_EQ(1, moving_average.GetAverageRoundedDown());
  73. EXPECT_EQ(1, moving_average.GetAverageRoundedToClosest());
  74. moving_average.Reset();
  75. EXPECT_EQ(0, moving_average.GetAverageRoundedDown());
  76. moving_average.AddSample(10);
  77. EXPECT_EQ(10, moving_average.GetAverageRoundedDown());
  78. EXPECT_EQ(10, moving_average.GetAverageRoundedToClosest());
  79. }
  80. TEST(MovingAverageTest, ManySamples) {
  81. MovingAverage moving_average(10);
  82. for (int i = 1; i < 11; i++) {
  83. moving_average.AddSample(i);
  84. }
  85. EXPECT_EQ(moving_average.GetAverageRoundedDown(), 5);
  86. EXPECT_EQ(moving_average.GetAverageRoundedToClosest(), 6);
  87. for (int i = 1; i < 2001; i++) {
  88. moving_average.AddSample(i);
  89. }
  90. EXPECT_EQ(moving_average.GetAverageRoundedDown(), 1995);
  91. EXPECT_EQ(moving_average.GetAverageRoundedToClosest(), 1996);
  92. }
  93. TEST(MovingAverageTest, VerifyNoOverflow) {
  94. constexpr int kMaxInt = std::numeric_limits<int>::max();
  95. MovingAverage moving_average(255);
  96. for (int i = 0; i < 255; i++) {
  97. moving_average.AddSample(kMaxInt);
  98. }
  99. EXPECT_EQ(moving_average.GetAverageRoundedDown(), kMaxInt);
  100. EXPECT_EQ(moving_average.GetAverageRoundedToClosest(), kMaxInt);
  101. EXPECT_EQ(moving_average.GetUnroundedAverage(), kMaxInt);
  102. }
  103. } // namespace test
  104. } // namespace base