1 | #include <memory> |
2 | |
3 | #include "../src/check.h" |
4 | #include "benchmark/benchmark.h" |
5 | #include "output_test.h" |
6 | |
7 | class TestMemoryManager : public benchmark::MemoryManager { |
8 | void Start() override {} |
9 | void Stop(Result& result) override { |
10 | result.num_allocs = 42; |
11 | result.max_bytes_used = 42000; |
12 | } |
13 | }; |
14 | |
15 | void BM_empty(benchmark::State& state) { |
16 | for (auto _ : state) { |
17 | auto iterations = double(state.iterations()) * double(state.iterations()); |
18 | benchmark::DoNotOptimize(value&: iterations); |
19 | } |
20 | } |
21 | BENCHMARK(BM_empty); |
22 | |
23 | ADD_CASES(TC_ConsoleOut, {{"^BM_empty %console_report$" }}); |
24 | ADD_CASES(TC_JSONOut, {{"\"name\": \"BM_empty\",$" }, |
25 | {"\"family_index\": 0,$" , MR_Next}, |
26 | {"\"per_family_instance_index\": 0,$" , MR_Next}, |
27 | {"\"run_name\": \"BM_empty\",$" , MR_Next}, |
28 | {"\"run_type\": \"iteration\",$" , MR_Next}, |
29 | {"\"repetitions\": 1,$" , MR_Next}, |
30 | {"\"repetition_index\": 0,$" , MR_Next}, |
31 | {"\"threads\": 1,$" , MR_Next}, |
32 | {"\"iterations\": %int,$" , MR_Next}, |
33 | {"\"real_time\": %float,$" , MR_Next}, |
34 | {"\"cpu_time\": %float,$" , MR_Next}, |
35 | {"\"time_unit\": \"ns\",$" , MR_Next}, |
36 | {"\"allocs_per_iter\": %float,$" , MR_Next}, |
37 | {"\"max_bytes_used\": 42000$" , MR_Next}, |
38 | {"}" , MR_Next}}); |
39 | ADD_CASES(TC_CSVOut, {{"^\"BM_empty\",%csv_report$" }}); |
40 | |
41 | int main(int argc, char* argv[]) { |
42 | std::unique_ptr<benchmark::MemoryManager> mm(new TestMemoryManager()); |
43 | |
44 | benchmark::RegisterMemoryManager(memory_manager: mm.get()); |
45 | RunOutputTests(argc, argv); |
46 | benchmark::RegisterMemoryManager(memory_manager: nullptr); |
47 | } |
48 | |