1/*
2 Copyright (c) Marshall Clow 2011-2012.
3
4 Distributed under the Boost Software License, Version 1.0. (See accompanying
5 file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
6
7 For more information, see http://www.boost.org
8*/
9
10#include <boost/config.hpp>
11#include <boost/algorithm/cxx11/iota.hpp>
12
13#define BOOST_TEST_MAIN
14#include <boost/test/unit_test.hpp>
15
16#include <iostream>
17#include <string>
18#include <vector>
19#include <list>
20
21// Test to make sure a sequence is "correctly formed"; i.e, ascending by one
22template <typename Iterator, typename T>
23BOOST_CXX14_CONSTEXPR bool test_iota_results ( Iterator first, Iterator last, T initial_value ) {
24 if ( first == last ) return true;
25 if ( initial_value != *first ) return false;
26 Iterator prev = first;
27 while ( ++first != last ) {
28 if (( *first - *prev ) != 1 )
29 return false;
30 prev = first;
31 }
32 return true;
33 }
34
35
36template <typename Range, typename T>
37BOOST_CXX14_CONSTEXPR bool test_iota_results ( const Range &r, T initial_value ) {
38 return test_iota_results (boost::begin (r), boost::end (r), initial_value );
39}
40
41
42void test_ints () {
43 std::vector<int> v;
44 std::list<int> l;
45
46 v.clear (); v.resize ( new_size: 10 );
47 boost::algorithm::iota ( first: v.begin (), last: v.end (), value: 23 );
48 BOOST_CHECK ( test_iota_results ( v.begin (), v.end (), 23 ));
49
50 v.clear (); v.resize ( new_size: 19 );
51 boost::algorithm::iota ( r&: v, value: 18 );
52 BOOST_CHECK ( test_iota_results ( v, 18 ));
53
54 v.clear ();
55 boost::algorithm::iota_n ( out: std::back_inserter(x&: v), value: 99, n: 20 );
56 BOOST_CHECK ( test_iota_results ( v, 99 ));
57
58 v.clear ();
59 boost::algorithm::iota_n ( out: std::back_inserter(x&: v), value: 99, n: 0 );
60 BOOST_CHECK ( v.size() == 0 );
61
62/*
63 l.clear (); l.reserve ( 5 );
64 boost::algorithm::iota ( l.begin (), l.end (), 123 );
65 BOOST_CHECK ( test_iota_results ( l.begin (), l.end (), 123 ));
66
67 l.clear (); l.reserve ( 9 );
68 boost::algorithm::iota ( l.begin (), l.end (), 87 );
69 BOOST_CHECK ( test_iota_results ( l.begin (), l.end (), 87 ));
70*/
71
72 l.clear ();
73 boost::algorithm::iota_n ( out: std::back_inserter(x&: l), value: 99, n: 20 );
74 BOOST_CHECK ( test_iota_results ( l, 99 ));
75
76 l.clear ();
77 boost::algorithm::iota_n ( out: std::front_inserter(x&: l), value: 123, n: 20 );
78 BOOST_CHECK ( test_iota_results ( l.rbegin (), l.rend (), 123 ));
79 }
80
81BOOST_CXX14_CONSTEXPR inline bool test_constexpr_iota() {
82 bool res = true;
83 int data[] = {0, 0, 0};
84 boost::algorithm::iota(first: data, last: data, value: 1); // fill none
85 res = (res && data[0] == 0);
86
87 boost::algorithm::iota(first: data, last: data + 3, value: 1); // fill all
88 res = (res && test_iota_results(first: data, last: data + 3, initial_value: 1));
89
90 return res;
91 }
92
93
94BOOST_CXX14_CONSTEXPR inline bool test_constexpr_iota_n() {
95 bool res = true;
96 int data[] = {0, 0, 0};
97 boost::algorithm::iota_n(out: data, value: 1, n: 0); // fill none
98 res = (res && data[0] == 0);
99
100 boost::algorithm::iota_n(out: data, value: 1, n: 3); // fill all
101 res = (res && test_iota_results(r: data, initial_value: 1));
102
103 return res;
104 }
105
106
107BOOST_AUTO_TEST_CASE( test_main )
108{
109 test_ints ();
110 BOOST_CXX14_CONSTEXPR bool constexpr_iota_res = test_constexpr_iota ();
111 BOOST_CHECK(constexpr_iota_res);
112 BOOST_CXX14_CONSTEXPR bool constexpr_iota_n_res = test_constexpr_iota_n ();
113 BOOST_CHECK(constexpr_iota_n_res);
114}
115

source code of boost/libs/algorithm/test/iota_test1.cpp