1 //  Copyright Neil Groves 2009. Use, modification and
2 //  distribution is subject to the Boost Software License, Version
3 //  1.0. (See accompanying file LICENSE_1_0.txt or copy at
4 //  http://www.boost.org/LICENSE_1_0.txt)
5 //
6 //
7 // For more information, see http://www.boost.org/libs/range/
8 //
9 #include <boost/range/algorithm/stable_partition.hpp>
10 
11 #include <boost/test/test_tools.hpp>
12 #include <boost/test/unit_test.hpp>
13 
14 #include <boost/assign.hpp>
15 #include "../test_driver/range_return_test_driver.hpp"
16 #include <algorithm>
17 #include <functional>
18 #include <list>
19 #include <numeric>
20 #include <deque>
21 #include <vector>
22 
23 namespace boost_range_test_algorithm_stable_partition
24 {
25     struct equal_to_5
26     {
27         typedef bool result_type;
28         typedef int argument_type;
operator ()boost_range_test_algorithm_stable_partition::equal_to_529         bool operator()(int x) const { return x == 5; }
30     };
31 
32     // test the 'partition' algorithm
33     template<class UnaryPredicate>
34     class stable_partition_test_policy
35     {
36     public:
37         template< class Container >
38         BOOST_DEDUCED_TYPENAME boost::range_iterator<Container>::type
test_iter(Container & cont)39         test_iter(Container& cont)
40         {
41             Container cont2(cont);
42 
43             typedef BOOST_DEDUCED_TYPENAME boost::range_iterator<Container>::type iter_t;
44             iter_t result = boost::stable_partition(cont, UnaryPredicate());
45 
46             iter_t temp_result = boost::stable_partition(
47                 boost::make_iterator_range(cont2), UnaryPredicate());
48 
49             BOOST_CHECK_EQUAL( std::distance(cont.begin(), result),
50                                std::distance(cont2.begin(), temp_result) );
51 
52             BOOST_CHECK_EQUAL_COLLECTIONS( cont.begin(), cont.end(),
53                                            cont2.begin(), cont2.end() );
54 
55             return result;
56         }
57 
pred() const58         UnaryPredicate pred() const { return UnaryPredicate(); }
59 
60         template< boost::range_return_value return_type >
61         struct test_range
62         {
63             template< class Container, class Policy >
64             BOOST_DEDUCED_TYPENAME boost::range_return<Container,return_type>::type
operator ()boost_range_test_algorithm_stable_partition::stable_partition_test_policy::test_range65             operator()(Policy& policy, Container& cont)
66             {
67                 typedef BOOST_DEDUCED_TYPENAME boost::range_return<Container,return_type>::type result_t;
68                 Container cont2(cont);
69                 result_t result = boost::stable_partition<return_type>(cont, policy.pred());
70 
71                 result_t result2 = boost::stable_partition<return_type>(
72                                     boost::make_iterator_range(cont2), policy.pred());
73 
74                 boost::ignore_unused_variable_warning(result2);
75 
76                 BOOST_CHECK_EQUAL_COLLECTIONS( cont2.begin(), cont2.end(),
77                                                cont.begin(), cont.end() );
78 
79                 return result;
80             }
81         };
82 
83         template< class Container >
84         BOOST_DEDUCED_TYPENAME boost::range_iterator<Container>::type
reference(Container & cont)85         reference(Container& cont)
86         {
87             return std::stable_partition(cont.begin(), cont.end(), UnaryPredicate());
88         }
89     };
90 
91     template<class Container>
test_stable_partition_impl()92     void test_stable_partition_impl()
93     {
94         using namespace boost::assign;
95 
96         boost::range_test::range_return_test_driver test_driver;
97 
98         stable_partition_test_policy< equal_to_5 > policy;
99 
100         Container cont;
101         test_driver(cont, policy);
102 
103         cont.clear();
104         cont += 1;
105         test_driver(cont, policy);
106 
107         cont.clear();
108         cont += 1,2,2,2,2,2,3,4,5,6,7,8,9;
109         test_driver(cont, policy);
110 
111         cont.clear();
112         cont += 1,2,2,2,2,2,3,3,3,3,4,4,4,4,4,4,4,5,6,7,8,9;
113         test_driver(cont, policy);
114     }
115 
test_stable_partition()116     void test_stable_partition()
117     {
118         test_stable_partition_impl< std::vector<int> >();
119         test_stable_partition_impl< std::list<int> >();
120         test_stable_partition_impl< std::deque<int> >();
121     }
122 }
123 
124 boost::unit_test::test_suite*
init_unit_test_suite(int argc,char * argv[])125 init_unit_test_suite(int argc, char* argv[])
126 {
127     boost::unit_test::test_suite* test
128         = BOOST_TEST_SUITE( "RangeTestSuite.algorithm.stable_partition" );
129 
130     test->add( BOOST_TEST_CASE( &boost_range_test_algorithm_stable_partition::test_stable_partition ) );
131 
132     return test;
133 }
134