utils/google-benchmark/test/multiple_ranges_test.cc

   1 #include "benchmark/benchmark.h"
   2
   3 #include <cassert>
   4 #include <set>
   5
   6 class MultipleRangesFixture : public ::benchmark::Fixture {
   7  public:
   8   MultipleRangesFixture()
   9       : expectedValues({{1, 3, 5},
  10                         {1, 3, 8},
  11                         {1, 3, 15},
  12                         {2, 3, 5},
  13                         {2, 3, 8},
  14                         {2, 3, 15},
  15                         {1, 4, 5},
  16                         {1, 4, 8},
  17                         {1, 4, 15},
  18                         {2, 4, 5},
  19                         {2, 4, 8},
  20                         {2, 4, 15},
  21                         {1, 7, 5},
  22                         {1, 7, 8},
  23                         {1, 7, 15},
  24                         {2, 7, 5},
  25                         {2, 7, 8},
  26                         {2, 7, 15},
  27                         {7, 6, 3}}) {}
  28
  29   void SetUp(const ::benchmark::State& state) {
  30     std::vector<int> ranges = {state.range(0), state.range(1), state.range(2)};
  31
  32     assert(expectedValues.find(ranges) != expectedValues.end());
  33
  34     actualValues.insert(ranges);
  35   }
  36
  37   virtual ~MultipleRangesFixture() {
  38     assert(actualValues.size() == expectedValues.size());
  39   }
  40
  41   std::set<std::vector<int>> expectedValues;
  42   std::set<std::vector<int>> actualValues;
  43 };
  44
  45 BENCHMARK_DEFINE_F(MultipleRangesFixture, Empty)(benchmark::State& state) {
  46   while (state.KeepRunning()) {
  47     int product = state.range(0) * state.range(1) * state.range(2);
  48     for (int x = 0; x < product; x++) {
  49       benchmark::DoNotOptimize(x);
  50     }
  51   }
  52 }
  53
  54 BENCHMARK_REGISTER_F(MultipleRangesFixture, Empty)
  55     ->RangeMultiplier(2)
  56     ->Ranges({{1, 2}, {3, 7}, {5, 15}})
  57     ->Args({7, 6, 3});
  58
  59 void BM_CheckDefaultArgument(benchmark::State& state) {
  60   // Test that the 'range()' without an argument is the same as 'range(0)'.
  61   assert(state.range() == state.range(0));
  62   assert(state.range() != state.range(1));
  63   while (state.KeepRunning()) {
  64   }
  65 }
  66 BENCHMARK(BM_CheckDefaultArgument)->Ranges({{1, 5}, {6, 10}});
  67
  68 static void BM_MultipleRanges(benchmark::State& st) {
  69   while (st.KeepRunning()) {
  70   }
  71 }
  72 BENCHMARK(BM_MultipleRanges)->Ranges({{5, 5}, {6, 6}});
  73
  74 BENCHMARK_MAIN()