| Index: third_party/google_benchmark/test/multiple_ranges_test.cc
|
| diff --git a/third_party/google_benchmark/test/multiple_ranges_test.cc b/third_party/google_benchmark/test/multiple_ranges_test.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..8e67b3b2a99cf62339e2118d663711b166b705b1
|
| --- /dev/null
|
| +++ b/third_party/google_benchmark/test/multiple_ranges_test.cc
|
| @@ -0,0 +1,74 @@
|
| +#include "benchmark/benchmark.h"
|
| +
|
| +#include <cassert>
|
| +#include <set>
|
| +
|
| +class MultipleRangesFixture : public ::benchmark::Fixture {
|
| + public:
|
| + MultipleRangesFixture()
|
| + : expectedValues({{1, 3, 5},
|
| + {1, 3, 8},
|
| + {1, 3, 15},
|
| + {2, 3, 5},
|
| + {2, 3, 8},
|
| + {2, 3, 15},
|
| + {1, 4, 5},
|
| + {1, 4, 8},
|
| + {1, 4, 15},
|
| + {2, 4, 5},
|
| + {2, 4, 8},
|
| + {2, 4, 15},
|
| + {1, 7, 5},
|
| + {1, 7, 8},
|
| + {1, 7, 15},
|
| + {2, 7, 5},
|
| + {2, 7, 8},
|
| + {2, 7, 15},
|
| + {7, 6, 3}}) {}
|
| +
|
| + void SetUp(const ::benchmark::State& state) {
|
| + std::vector<int> ranges = {state.range(0), state.range(1), state.range(2)};
|
| +
|
| + assert(expectedValues.find(ranges) != expectedValues.end());
|
| +
|
| + actualValues.insert(ranges);
|
| + }
|
| +
|
| + virtual ~MultipleRangesFixture() {
|
| + assert(actualValues.size() == expectedValues.size());
|
| + }
|
| +
|
| + std::set<std::vector<int>> expectedValues;
|
| + std::set<std::vector<int>> actualValues;
|
| +};
|
| +
|
| +BENCHMARK_DEFINE_F(MultipleRangesFixture, Empty)(benchmark::State& state) {
|
| + while (state.KeepRunning()) {
|
| + int product = state.range(0) * state.range(1) * state.range(2);
|
| + for (int x = 0; x < product; x++) {
|
| + benchmark::DoNotOptimize(x);
|
| + }
|
| + }
|
| +}
|
| +
|
| +BENCHMARK_REGISTER_F(MultipleRangesFixture, Empty)
|
| + ->RangeMultiplier(2)
|
| + ->Ranges({{1, 2}, {3, 7}, {5, 15}})
|
| + ->Args({7, 6, 3});
|
| +
|
| +void BM_CheckDefaultArgument(benchmark::State& state) {
|
| + // Test that the 'range()' without an argument is the same as 'range(0)'.
|
| + assert(state.range() == state.range(0));
|
| + assert(state.range() != state.range(1));
|
| + while (state.KeepRunning()) {
|
| + }
|
| +}
|
| +BENCHMARK(BM_CheckDefaultArgument)->Ranges({{1, 5}, {6, 10}});
|
| +
|
| +static void BM_MultipleRanges(benchmark::State& st) {
|
| + while (st.KeepRunning()) {
|
| + }
|
| +}
|
| +BENCHMARK(BM_MultipleRanges)->Ranges({{5, 5}, {6, 6}});
|
| +
|
| +BENCHMARK_MAIN()
|
|
|