1#include <cassert>
2#include <iostream>
3#include <set>
4#include <vector>
5
6#include "benchmark/benchmark.h"
7
8class MultipleRangesFixture : public ::benchmark::Fixture {
9 public:
10  MultipleRangesFixture()
11      : expectedValues({{1, 3, 5},
12                        {1, 3, 8},
13                        {1, 3, 15},
14                        {2, 3, 5},
15                        {2, 3, 8},
16                        {2, 3, 15},
17                        {1, 4, 5},
18                        {1, 4, 8},
19                        {1, 4, 15},
20                        {2, 4, 5},
21                        {2, 4, 8},
22                        {2, 4, 15},
23                        {1, 7, 5},
24                        {1, 7, 8},
25                        {1, 7, 15},
26                        {2, 7, 5},
27                        {2, 7, 8},
28                        {2, 7, 15},
29                        {7, 6, 3}}) {}
30
31  void SetUp(const ::benchmark::State& state) override {
32    std::vector<int64_t> ranges = {state.range(0), state.range(1),
33                                   state.range(2)};
34
35    assert(expectedValues.find(ranges) != expectedValues.end());
36
37    actualValues.insert(ranges);
38  }
39
40  // NOTE: This is not TearDown as we want to check after _all_ runs are
41  // complete.
42  ~MultipleRangesFixture() override {
43    if (actualValues != expectedValues) {
44      std::cout << "EXPECTED\n";
45      for (const auto& v : expectedValues) {
46        std::cout << "{";
47        for (int64_t iv : v) {
48          std::cout << iv << ", ";
49        }
50        std::cout << "}\n";
51      }
52      std::cout << "ACTUAL\n";
53      for (const auto& v : actualValues) {
54        std::cout << "{";
55        for (int64_t iv : v) {
56          std::cout << iv << ", ";
57        }
58        std::cout << "}\n";
59      }
60    }
61  }
62
63  std::set<std::vector<int64_t>> expectedValues;
64  std::set<std::vector<int64_t>> actualValues;
65};
66
67BENCHMARK_DEFINE_F(MultipleRangesFixture, Empty)(benchmark::State& state) {
68  for (auto _ : state) {
69    int64_t product = state.range(0) * state.range(1) * state.range(2);
70    for (int64_t x = 0; x < product; x++) {
71      benchmark::DoNotOptimize(x);
72    }
73  }
74}
75
76BENCHMARK_REGISTER_F(MultipleRangesFixture, Empty)
77    ->RangeMultiplier(2)
78    ->Ranges({{1, 2}, {3, 7}, {5, 15}})
79    ->Args({7, 6, 3});
80
81void BM_CheckDefaultArgument(benchmark::State& state) {
82  // Test that the 'range()' without an argument is the same as 'range(0)'.
83  assert(state.range() == state.range(0));
84  assert(state.range() != state.range(1));
85  for (auto _ : state) {
86  }
87}
88BENCHMARK(BM_CheckDefaultArgument)->Ranges({{1, 5}, {6, 10}});
89
90static void BM_MultipleRanges(benchmark::State& st) {
91  for (auto _ : st) {
92  }
93}
94BENCHMARK(BM_MultipleRanges)->Ranges({{5, 5}, {6, 6}});
95
96BENCHMARK_MAIN();
97