utils/benchmark/test/fixture_test.cc

   1
   2 #include "benchmark/benchmark.h"
   3
   4 #include <cassert>
   5 #include <memory>
   6
   7 class MyFixture : public ::benchmark::Fixture {
   8  public:
   9   void SetUp(const ::benchmark::State& state) {
  10     if (state.thread_index == 0) {
  11       assert(data.get() == nullptr);
  12       data.reset(new int(42));
  13     }
  14   }
  15
  16   void TearDown(const ::benchmark::State& state) {
  17     if (state.thread_index == 0) {
  18       assert(data.get() != nullptr);
  19       data.reset();
  20     }
  21   }
  22
  23   ~MyFixture() { assert(data == nullptr); }
  24
  25   std::unique_ptr<int> data;
  26 };
  27
  28 BENCHMARK_F(MyFixture, Foo)(benchmark::State &st) {
  29   assert(data.get() != nullptr);
  30   assert(*data == 42);
  31   for (auto _ : st) {
  32   }
  33 }
  34
  35 BENCHMARK_DEFINE_F(MyFixture, Bar)(benchmark::State& st) {
  36   if (st.thread_index == 0) {
  37     assert(data.get() != nullptr);
  38     assert(*data == 42);
  39   }
  40   for (auto _ : st) {
  41     assert(data.get() != nullptr);
  42     assert(*data == 42);
  43   }
  44   st.SetItemsProcessed(st.range(0));
  45 }
  46 BENCHMARK_REGISTER_F(MyFixture, Bar)->Arg(42);
  47 BENCHMARK_REGISTER_F(MyFixture, Bar)->Arg(42)->ThreadPerCpu();
  48
  49 BENCHMARK_MAIN();