diff --git a/benchmark/include/utility/io/generate.hpp b/benchmark/include/utility/io/generate.hpp index 22dd6927..0193fc72 100644 --- a/benchmark/include/utility/io/generate.hpp +++ b/benchmark/include/utility/io/generate.hpp @@ -8,15 +8,16 @@ #include #include #include -#include +#include #include #include namespace utility { - void + std::ostream & generate ( + std::ostream & stream, std::size_t range_count, std::size_t range_length, std::int64_t min, @@ -51,8 +52,10 @@ namespace utility } } - write(std::cout, range); + write(stream, range); } + + return stream; } } // namespace utility diff --git a/benchmark/src/range_generator.cpp b/benchmark/src/range_generator.cpp index b440f523..1c5f5a91 100644 --- a/benchmark/src/range_generator.cpp +++ b/benchmark/src/range_generator.cpp @@ -51,7 +51,7 @@ int main (int argc, const char * argv[]) bool sort = vm["sort"].as(); bool descending = vm["descending"].as(); - utility::generate(range_count, range_length, min, max, seed, sort, descending); + utility::generate(std::cout, range_count, range_length, min, max, seed, sort, descending); } } catch (std::exception &)