mfbt/tests/TestRollingMean.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
   5  * You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "mozilla/Assertions.h"
   8 #include "mozilla/RollingMean.h"
   9
  10 using mozilla::RollingMean;
  11
  12 class MyClass
  13 {
  14 public:
  15   uint32_t mValue;
  16
  17   explicit MyClass(uint32_t aValue = 0) : mValue(aValue) { }
  18
  19   bool operator==(const MyClass& aOther) const
  20   {
  21     return mValue == aOther.mValue;
  22   }
  23
  24   MyClass operator+(const MyClass& aOther) const
  25   {
  26     return MyClass(mValue + aOther.mValue);
  27   }
  28
  29   MyClass operator-(const MyClass& aOther) const
  30   {
  31     return MyClass(mValue - aOther.mValue);
  32   }
  33
  34   MyClass operator/(uint32_t aDiv) const
  35   {
  36     return MyClass(mValue / aDiv);
  37   }
  38 };
  39
  40 class RollingMeanSuite
  41 {
  42 public:
  43   RollingMeanSuite() { }
  44
  45   void runTests() {
  46     testZero();
  47     testClear();
  48     testRolling();
  49     testClass();
  50     testMove();
  51   }
  52
  53 private:
  54   void testZero()
  55   {
  56     RollingMean<uint32_t, uint64_t> mean(3);
  57     MOZ_RELEASE_ASSERT(mean.empty());
  58   }
  59
  60   void testClear()
  61   {
  62     RollingMean<uint32_t, uint64_t> mean(3);
  63
  64     mean.insert(4);
  65     MOZ_RELEASE_ASSERT(mean.mean() == 4);
  66
  67     mean.clear();
  68     MOZ_RELEASE_ASSERT(mean.empty());
  69
  70     mean.insert(3);
  71     MOZ_RELEASE_ASSERT(mean.mean() == 3);
  72   }
  73
  74   void testRolling()
  75   {
  76     RollingMean<uint32_t, uint64_t> mean(3);
  77
  78     mean.insert(10);
  79     MOZ_RELEASE_ASSERT(mean.mean() == 10);
  80
  81     mean.insert(20);
  82     MOZ_RELEASE_ASSERT(mean.mean() == 15);
  83
  84     mean.insert(35);
  85     MOZ_RELEASE_ASSERT(mean.mean() == 21);
  86
  87     mean.insert(5);
  88     MOZ_RELEASE_ASSERT(mean.mean() == 20);
  89
  90     mean.insert(10);
  91     MOZ_RELEASE_ASSERT(mean.mean() == 16);
  92   }
  93
  94   void testClass()
  95   {
  96     RollingMean<MyClass, MyClass> mean(3);
  97
  98     mean.insert(MyClass(4));
  99     MOZ_RELEASE_ASSERT(mean.mean() == MyClass(4));
 100
 101     mean.clear();
 102     MOZ_RELEASE_ASSERT(mean.empty());
 103   }
 104
 105   void testMove()
 106   {
 107     RollingMean<uint32_t, uint64_t> mean(3);
 108     mean = RollingMean<uint32_t, uint64_t>(4);
 109     MOZ_RELEASE_ASSERT(mean.maxValues() == 4);
 110
 111     mean.insert(10);
 112     MOZ_RELEASE_ASSERT(mean.mean() == 10);
 113
 114     mean = RollingMean<uint32_t, uint64_t>(3);
 115     mean.insert(30);
 116     mean.insert(40);
 117     mean.insert(50);
 118     mean.insert(60);
 119     MOZ_RELEASE_ASSERT(mean.mean() == 50);
 120   }
 121 };
 122
 123 int
 124 main()
 125 {
 126   RollingMeanSuite suite;
 127   suite.runTests();
 128   return 0;
 129 }