utils/benchmark/test/options_test.cc

   1 #include "benchmark/benchmark.h"
   2 #include <chrono>
   3 #include <thread>
   4
   5 #if defined(NDEBUG)
   6 #undef NDEBUG
   7 #endif
   8 #include <cassert>
   9
  10 void BM_basic(benchmark::State& state) {
  11   for (auto _ : state) {
  12   }
  13 }
  14
  15 void BM_basic_slow(benchmark::State& state) {
  16   std::chrono::milliseconds sleep_duration(state.range(0));
  17   for (auto _ : state) {
  18     std::this_thread::sleep_for(
  19         std::chrono::duration_cast<std::chrono::nanoseconds>(sleep_duration));
  20   }
  21 }
  22
  23 BENCHMARK(BM_basic);
  24 BENCHMARK(BM_basic)->Arg(42);
  25 BENCHMARK(BM_basic_slow)->Arg(10)->Unit(benchmark::kNanosecond);
  26 BENCHMARK(BM_basic_slow)->Arg(100)->Unit(benchmark::kMicrosecond);
  27 BENCHMARK(BM_basic_slow)->Arg(1000)->Unit(benchmark::kMillisecond);
  28 BENCHMARK(BM_basic)->Range(1, 8);
  29 BENCHMARK(BM_basic)->RangeMultiplier(2)->Range(1, 8);
  30 BENCHMARK(BM_basic)->DenseRange(10, 15);
  31 BENCHMARK(BM_basic)->Args({42, 42});
  32 BENCHMARK(BM_basic)->Ranges({{64, 512}, {64, 512}});
  33 BENCHMARK(BM_basic)->MinTime(0.7);
  34 BENCHMARK(BM_basic)->UseRealTime();
  35 BENCHMARK(BM_basic)->ThreadRange(2, 4);
  36 BENCHMARK(BM_basic)->ThreadPerCpu();
  37 BENCHMARK(BM_basic)->Repetitions(3);
  38
  39 void CustomArgs(benchmark::internal::Benchmark* b) {
  40   for (int i = 0; i < 10; ++i) {
  41     b->Arg(i);
  42   }
  43 }
  44
  45 BENCHMARK(BM_basic)->Apply(CustomArgs);
  46
  47 void BM_explicit_iteration_count(benchmark::State& state) {
  48   // Test that benchmarks specified with an explicit iteration count are
  49   // only run once.
  50   static bool invoked_before = false;
  51   assert(!invoked_before);
  52   invoked_before = true;
  53
  54   // Test that the requested iteration count is respected.
  55   assert(state.max_iterations == 42);
  56   size_t actual_iterations = 0;
  57   for (auto _ : state)
  58     ++actual_iterations;
  59   assert(state.iterations() == state.max_iterations);
  60   assert(state.iterations() == 42);
  61
  62 }
  63 BENCHMARK(BM_explicit_iteration_count)->Iterations(42);
  64
  65 BENCHMARK_MAIN();