From 1dd40c99c092e16488387c3409a6f8600a406d00 Mon Sep 17 00:00:00 2001
From: Eli Bendersky <eliben@gmail.com>
Date: Wed, 23 Sep 2015 11:44:26 -0700
Subject: [PATCH] Custom arguments function usage with Apply - fix doc and add
 test.

---
 include/benchmark/benchmark_api.h | 3 +--
 test/options_test.cc              | 8 ++++++++
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/include/benchmark/benchmark_api.h b/include/benchmark/benchmark_api.h
index 4d9d5f3e..f098255d 100644
--- a/include/benchmark/benchmark_api.h
+++ b/include/benchmark/benchmark_api.h
@@ -90,8 +90,7 @@ BENCHMARK(BM_SetInsert)->RangePair(1<<10, 8<<10, 1, 512);
 static void CustomArguments(benchmark::internal::Benchmark* b) {
   for (int i = 0; i <= 10; ++i)
     for (int j = 32; j <= 1024*1024; j *= 8)
-      b = b->ArgPair(i, j);
-  return b;
+      b->ArgPair(i, j);
 }
 BENCHMARK(BM_SetInsert)->Apply(CustomArguments);
 
diff --git a/test/options_test.cc b/test/options_test.cc
index ec8c2a16..d4c682d4 100644
--- a/test/options_test.cc
+++ b/test/options_test.cc
@@ -15,4 +15,12 @@ BENCHMARK(BM_basic)->UseRealTime();
 BENCHMARK(BM_basic)->ThreadRange(2, 4);
 BENCHMARK(BM_basic)->ThreadPerCpu();
 
+void CustomArgs(benchmark::internal::Benchmark* b) {
+  for (int i = 0; i < 10; ++i) {
+    b->Arg(i);
+  }
+}
+
+BENCHMARK(BM_basic)->Apply(CustomArgs);
+
 BENCHMARK_MAIN()