libs/utility/generator_iterator_test.cpp

   1 //
   2 // Copyright 2014 Peter Dimov
   3 //
   4 // Distributed under the Boost Software License, Version 1.0.
   5 // See accompanying file LICENSE_1_0.txt or copy at
   6 // http://www.boost.org/LICENSE_1_0.txt
   7 //
   8
   9 #include <boost/generator_iterator.hpp>
  10 #include <boost/detail/lightweight_test.hpp>
  11 #include <algorithm>
  12
  13 class X
  14 {
  15 private:
  16
  17     int v;
  18
  19 public:
  20
  21     typedef int result_type;
  22
  23     X(): v( 0 )
  24     {
  25     }
  26
  27     int operator()()
  28     {
  29         return ++v;
  30     }
  31 };
  32
  33 template<class InputIterator, class Size, class OutputIterator> OutputIterator copy_n( InputIterator first, Size n, OutputIterator result )
  34 {
  35     while( n-- > 0 )
  36     {
  37         *result++ = *first++;
  38     }
  39
  40     return result;
  41 }
  42
  43 void copy_test()
  44 {
  45     X x;
  46     boost::generator_iterator<X> in( &x );
  47
  48     int const N = 4;
  49     int v[ N ] = { 0 };
  50
  51     ::copy_n( in, 4, v );
  52
  53     BOOST_TEST_EQ( v[0], 1 );
  54     BOOST_TEST_EQ( v[1], 2 );
  55     BOOST_TEST_EQ( v[2], 3 );
  56     BOOST_TEST_EQ( v[3], 4 );
  57 }
  58
  59 int main()
  60 {
  61     copy_test();
  62     return boost::report_errors();
  63 }