libs/iterator/test/function_input_iterator_test.cpp

   1 // Copyright 2010 (c) Dean Michael Berris
   2 // Distributed under the Boost Software License Version 1.0.
   3 // (See accompanying file LICENSE_1_0.txt or copy at
   4 // http://www.boost.org/LICENSE_1_0.txt)
   5
   6 #include <cassert>
   7 #include <cstddef>
   8
   9 #include <algorithm>
  10 #include <iostream>
  11 #include <iterator>
  12 #include <vector>
  13
  14 #include <boost/config.hpp>
  15 #include <boost/iterator/function_input_iterator.hpp>
  16
  17 namespace {
  18
  19 struct ones {
  20     typedef int result_type;
  21     result_type operator() () {
  22         return 1;
  23     }
  24 };
  25
  26 int ones_function () {
  27     return 1;
  28 }
  29
  30 struct counter {
  31     typedef int result_type;
  32     int n;
  33     explicit counter(int n_) : n(n_) { }
  34     result_type operator() () {
  35         return n++;
  36     }
  37 };
  38
  39 } // namespace
  40
  41 using namespace std;
  42
  43 int main(int argc, char * argv[])
  44 {
  45     // test the iterator with function objects
  46     ones ones_generator;
  47     vector<int> values(10);
  48     generate(values.begin(), values.end(), ones());
  49
  50     vector<int> generated;
  51     copy(
  52         boost::make_function_input_iterator(ones_generator, 0),
  53         boost::make_function_input_iterator(ones_generator, 10),
  54         back_inserter(generated)
  55         );
  56
  57     assert(values.size() == generated.size());
  58     assert(equal(values.begin(), values.end(), generated.begin()));
  59     cout << "function iterator test with function objects successful." << endl;
  60
  61     // test the iterator with normal functions
  62     vector<int>().swap(generated);
  63     copy(
  64         boost::make_function_input_iterator(&ones_function, 0),
  65         boost::make_function_input_iterator(&ones_function, 10),
  66         back_inserter(generated)
  67         );
  68
  69     assert(values.size() == generated.size());
  70     assert(equal(values.begin(), values.end(), generated.begin()));
  71     cout << "function iterator test with pointer to function successful." << endl;
  72
  73     // test the iterator with a reference to a function
  74     vector<int>().swap(generated);
  75     copy(
  76         boost::make_function_input_iterator(ones_function, 0),
  77         boost::make_function_input_iterator(ones_function, 10),
  78         back_inserter(generated)
  79         );
  80
  81     assert(values.size() == generated.size());
  82     assert(equal(values.begin(), values.end(), generated.begin()));
  83     cout << "function iterator test with reference to function successful." << endl;
  84
  85     // test the iterator with a stateful function object
  86     counter counter_generator(42);
  87     vector<int>().swap(generated);
  88     copy(
  89         boost::make_function_input_iterator(counter_generator, 0),
  90         boost::make_function_input_iterator(counter_generator, 10),
  91         back_inserter(generated)
  92         );
  93
  94     assert(generated.size() == 10);
  95     assert(counter_generator.n == 42 + 10);
  96     for(std::size_t i = 0; i != 10; ++i)
  97         assert(generated[i] == 42 + i);
  98     cout << "function iterator test with stateful function object successful." << endl;
  99
 100 #if !defined(BOOST_NO_CXX11_LAMBDAS) && !defined(BOOST_NO_CXX11_AUTO_DECLARATIONS)
 101     // test the iterator with lambda expressions
 102     int num = 42;
 103     auto lambda_generator = [&num] { return num++; };
 104     vector<int>().swap(generated);
 105     copy(
 106         boost::make_function_input_iterator(lambda_generator, 0),
 107         boost::make_function_input_iterator(lambda_generator, 10),
 108         back_inserter(generated)
 109         );
 110
 111     assert(generated.size() == 10);
 112     for(std::size_t i = 0; i != 10; ++i)
 113         assert(generated[i] == 42 + i);
 114     cout << "function iterator test with lambda expressions successful." << endl;
 115 #endif // BOOST_NO_CXX11_LAMBDAS
 116
 117     return 0;
 118 }
 119