1 | // Boost.Range library |
2 | // |
3 | // Copyright Neil Groves 2009. Use, modification and |
4 | // distribution is subject to the Boost Software License, Version |
5 | // 1.0. (See accompanying file LICENSE_1_0.txt or copy at |
6 | // http://www.boost.org/LICENSE_1_0.txt) |
7 | // |
8 | // |
9 | // For more information, see http://www.boost.org/libs/range/ |
10 | // |
11 | #include <boost/range/adaptor/transformed.hpp> |
12 | |
13 | #include <boost/test/test_tools.hpp> |
14 | #include <boost/test/unit_test.hpp> |
15 | |
16 | #include <boost/assign.hpp> |
17 | #include <boost/bind/bind.hpp> |
18 | #include <boost/range/algorithm_ext.hpp> |
19 | |
20 | #include <algorithm> |
21 | #include <list> |
22 | #include <set> |
23 | #include <vector> |
24 | |
25 | namespace boost |
26 | { |
27 | namespace |
28 | { |
29 | struct double_x |
30 | { |
31 | typedef int result_type; |
32 | int operator()(int x) const { return x * 2; } |
33 | }; |
34 | |
35 | struct halve_x |
36 | { |
37 | typedef int result_type; |
38 | int operator()(int x) const { return x / 2; } |
39 | }; |
40 | |
41 | struct lambda_init |
42 | { |
43 | }; |
44 | |
45 | struct lambda |
46 | { |
47 | typedef int result_type; |
48 | |
49 | lambda(const lambda_init& init) {} |
50 | lambda(const lambda& rhs) {} |
51 | |
52 | int operator()(int x) const { return x + 1; } |
53 | |
54 | private: |
55 | lambda() {} |
56 | lambda& operator=(const lambda& rhs) { return *this; } |
57 | }; |
58 | |
59 | template< class Container, class TransformFn > |
60 | void transformed_test_impl_core( Container& c, TransformFn fn ) |
61 | { |
62 | using namespace boost::adaptors; |
63 | |
64 | std::vector< int > test_result1; |
65 | boost::push_back(test_result1, c | transformed(fn)); |
66 | |
67 | std::vector< int > test_result2; |
68 | boost::push_back(test_result2, adaptors::transform(c, fn)); |
69 | |
70 | std::vector< int > reference; |
71 | std::transform(c.begin(), c.end(), std::back_inserter(x&: reference), fn); |
72 | |
73 | BOOST_CHECK_EQUAL_COLLECTIONS( reference.begin(), reference.end(), |
74 | test_result1.begin(), test_result1.end() ); |
75 | |
76 | BOOST_CHECK_EQUAL_COLLECTIONS( reference.begin(), reference.end(), |
77 | test_result2.begin(), test_result2.end() ); |
78 | } |
79 | |
80 | template< class Rng > |
81 | void check_copy_assign(Rng r) |
82 | { |
83 | Rng r2 = r; |
84 | r2 = r; |
85 | } |
86 | |
87 | template< class Container, class TransformFn > |
88 | void transformed_range_copy_assign(Container& c, TransformFn fn) |
89 | { |
90 | using namespace boost::adaptors; |
91 | check_copy_assign(c | transformed(fn)); |
92 | check_copy_assign(adaptors::transform(c, fn)); |
93 | } |
94 | |
95 | template< class Container, class TransformFn, class TransformFnInit > |
96 | void transformed_test_fn_impl() |
97 | { |
98 | using namespace boost::assign; |
99 | |
100 | Container c; |
101 | TransformFnInit init; |
102 | TransformFn fn( init ); |
103 | |
104 | // Test empty |
105 | transformed_test_impl_core(c, fn); |
106 | |
107 | // Test one element |
108 | c += 1; |
109 | transformed_test_impl_core(c, fn); |
110 | |
111 | // Test many elements |
112 | c += 1,1,1,2,2,2,2,2,3,4,5,6,7,8,9; |
113 | transformed_test_impl_core(c, fn); |
114 | |
115 | // test the range and iterator are copy assignable |
116 | transformed_range_copy_assign(c, fn); |
117 | } |
118 | |
119 | template< class Container > |
120 | void transformed_test_impl() |
121 | { |
122 | transformed_test_fn_impl< Container, double_x, double_x >(); |
123 | transformed_test_fn_impl< Container, halve_x, halve_x >(); |
124 | transformed_test_fn_impl< Container, lambda, lambda_init >(); |
125 | } |
126 | |
127 | void transformed_test() |
128 | { |
129 | transformed_test_impl< std::vector< int > >(); |
130 | transformed_test_impl< std::list< int > >(); |
131 | transformed_test_impl< std::set< int > >(); |
132 | transformed_test_impl< std::multiset< int > >(); |
133 | } |
134 | |
135 | struct foo_bind |
136 | { |
137 | int foo() const { return 7; } |
138 | }; |
139 | |
140 | void transformed_bind() |
141 | { |
142 | using namespace boost::adaptors; |
143 | using namespace boost::placeholders; |
144 | |
145 | std::vector<foo_bind> input(5); |
146 | std::vector<int> output; |
147 | boost::range::push_back( |
148 | on&: output, |
149 | from: input | transformed(boost::bind(f: &foo_bind::foo, a1: _1))); |
150 | |
151 | BOOST_CHECK_EQUAL(output.size(), input.size()); |
152 | |
153 | std::vector<int> reference_output(5, 7); |
154 | BOOST_CHECK_EQUAL_COLLECTIONS( |
155 | output.begin(), output.end(), |
156 | reference_output.begin(), reference_output.end()); |
157 | } |
158 | } |
159 | } |
160 | |
161 | boost::unit_test::test_suite* |
162 | init_unit_test_suite(int argc, char* argv[]) |
163 | { |
164 | boost::unit_test::test_suite* test |
165 | = BOOST_TEST_SUITE( "RangeTestSuite.adaptor.transformed" ); |
166 | |
167 | test->add(BOOST_TEST_CASE(&boost::transformed_test)); |
168 | test->add(BOOST_TEST_CASE(&boost::transformed_bind)); |
169 | |
170 | return test; |
171 | } |
172 | |