123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170 |
- // Boost.Range library
- //
- // Copyright Neil Groves 2009. Use, modification and
- // distribution is subject to the Boost Software License, Version
- // 1.0. (See accompanying file LICENSE_1_0.txt or copy at
- // http://www.boost.org/LICENSE_1_0.txt)
- //
- //
- // For more information, see http://www.boost.org/libs/range/
- //
- #include <boost/range/adaptor/transformed.hpp>
- #include <boost/test/test_tools.hpp>
- #include <boost/test/unit_test.hpp>
- #include <boost/assign.hpp>
- #include <boost/bind.hpp>
- #include <boost/range/algorithm_ext.hpp>
- #include <algorithm>
- #include <list>
- #include <set>
- #include <vector>
- namespace boost
- {
- namespace
- {
- struct double_x
- {
- typedef int result_type;
- int operator()(int x) const { return x * 2; }
- };
- struct halve_x
- {
- typedef int result_type;
- int operator()(int x) const { return x / 2; }
- };
- struct lambda_init
- {
- };
- struct lambda
- {
- typedef int result_type;
- lambda(const lambda_init& init) {}
- lambda(const lambda& rhs) {}
- int operator()(int x) const { return x + 1; }
- private:
- lambda() {}
- lambda& operator=(const lambda& rhs) { return *this; }
- };
- template< class Container, class TransformFn >
- void transformed_test_impl_core( Container& c, TransformFn fn )
- {
- using namespace boost::adaptors;
- std::vector< int > test_result1;
- boost::push_back(test_result1, c | transformed(fn));
- std::vector< int > test_result2;
- boost::push_back(test_result2, adaptors::transform(c, fn));
- std::vector< int > reference;
- std::transform(c.begin(), c.end(), std::back_inserter(reference), fn);
- BOOST_CHECK_EQUAL_COLLECTIONS( reference.begin(), reference.end(),
- test_result1.begin(), test_result1.end() );
- BOOST_CHECK_EQUAL_COLLECTIONS( reference.begin(), reference.end(),
- test_result2.begin(), test_result2.end() );
- }
- template< class Rng >
- void check_copy_assign(Rng r)
- {
- Rng r2 = r;
- r2 = r;
- }
- template< class Container, class TransformFn >
- void transformed_range_copy_assign(Container& c, TransformFn fn)
- {
- using namespace boost::adaptors;
- check_copy_assign(c | transformed(fn));
- check_copy_assign(adaptors::transform(c, fn));
- }
- template< class Container, class TransformFn, class TransformFnInit >
- void transformed_test_fn_impl()
- {
- using namespace boost::assign;
- Container c;
- TransformFnInit init;
- TransformFn fn( init );
- // Test empty
- transformed_test_impl_core(c, fn);
- // Test one element
- c += 1;
- transformed_test_impl_core(c, fn);
- // Test many elements
- c += 1,1,1,2,2,2,2,2,3,4,5,6,7,8,9;
- transformed_test_impl_core(c, fn);
- // test the range and iterator are copy assignable
- transformed_range_copy_assign(c, fn);
- }
- template< class Container >
- void transformed_test_impl()
- {
- transformed_test_fn_impl< Container, double_x, double_x >();
- transformed_test_fn_impl< Container, halve_x, halve_x >();
- transformed_test_fn_impl< Container, lambda, lambda_init >();
- }
- void transformed_test()
- {
- transformed_test_impl< std::vector< int > >();
- transformed_test_impl< std::list< int > >();
- transformed_test_impl< std::set< int > >();
- transformed_test_impl< std::multiset< int > >();
- }
- struct foo_bind
- {
- int foo() const { return 7; }
- };
- void transformed_bind()
- {
- using namespace boost::adaptors;
- std::vector<foo_bind> input(5);
- std::vector<int> output;
- boost::range::push_back(
- output,
- input | transformed(boost::bind(&foo_bind::foo, _1)));
- BOOST_CHECK_EQUAL(output.size(), input.size());
- std::vector<int> reference_output(5, 7);
- BOOST_CHECK_EQUAL_COLLECTIONS(
- output.begin(), output.end(),
- reference_output.begin(), reference_output.end());
- }
- }
- }
- boost::unit_test::test_suite*
- init_unit_test_suite(int argc, char* argv[])
- {
- boost::unit_test::test_suite* test
- = BOOST_TEST_SUITE( "RangeTestSuite.adaptor.transformed" );
- test->add(BOOST_TEST_CASE(&boost::transformed_test));
- test->add(BOOST_TEST_CASE(&boost::transformed_bind));
- return test;
- }
|