merge.cpp 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237
  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/merge.hpp>
  10. #include <boost/test/test_tools.hpp>
  11. #include <boost/test/unit_test.hpp>
  12. #include <boost/assign.hpp>
  13. #include <boost/bind.hpp>
  14. #include <algorithm>
  15. #include <functional>
  16. #include <list>
  17. #include <numeric>
  18. #include <deque>
  19. #include <vector>
  20. namespace boost
  21. {
  22. namespace
  23. {
  24. template<class Container1, class Container2>
  25. void test(Container1& cont1, Container2& cont2)
  26. {
  27. typedef BOOST_DEDUCED_TYPENAME Container1::value_type value_t;
  28. typedef BOOST_DEDUCED_TYPENAME std::vector<value_t>::iterator iterator_t;
  29. std::vector<value_t> reference_target( cont1.size() + cont2.size() );
  30. iterator_t reference_it
  31. = std::merge(cont1.begin(), cont1.end(),
  32. cont2.begin(), cont2.end(),
  33. reference_target.begin());
  34. std::vector<value_t> test_target( cont1.size() + cont2.size() );
  35. iterator_t test_it
  36. = boost::merge(cont1, cont2, test_target.begin());
  37. BOOST_CHECK_EQUAL(
  38. std::distance<iterator_t>(reference_target.begin(), reference_it),
  39. std::distance<iterator_t>(test_target.begin(), test_it)
  40. );
  41. BOOST_CHECK_EQUAL_COLLECTIONS(
  42. reference_target.begin(), reference_target.end(),
  43. test_target.begin(), test_target.end()
  44. );
  45. test_it = boost::merge(boost::make_iterator_range(cont1),
  46. cont2, test_target.begin());
  47. BOOST_CHECK_EQUAL(
  48. std::distance<iterator_t>(reference_target.begin(), reference_it),
  49. std::distance<iterator_t>(test_target.begin(), test_it)
  50. );
  51. BOOST_CHECK_EQUAL_COLLECTIONS(
  52. reference_target.begin(), reference_target.end(),
  53. test_target.begin(), test_target.end()
  54. );
  55. test_it = boost::merge(cont1, boost::make_iterator_range(cont2),
  56. test_target.begin());
  57. BOOST_CHECK_EQUAL(
  58. std::distance<iterator_t>(reference_target.begin(), reference_it),
  59. std::distance<iterator_t>(test_target.begin(), test_it)
  60. );
  61. BOOST_CHECK_EQUAL_COLLECTIONS(
  62. reference_target.begin(), reference_target.end(),
  63. test_target.begin(), test_target.end()
  64. );
  65. test_it = boost::merge(boost::make_iterator_range(cont1),
  66. boost::make_iterator_range(cont2),
  67. test_target.begin());
  68. BOOST_CHECK_EQUAL(
  69. std::distance<iterator_t>(reference_target.begin(), reference_it),
  70. std::distance<iterator_t>(test_target.begin(), test_it)
  71. );
  72. BOOST_CHECK_EQUAL_COLLECTIONS(
  73. reference_target.begin(), reference_target.end(),
  74. test_target.begin(), test_target.end()
  75. );
  76. }
  77. template<class Container, class BinaryPredicate>
  78. void sort_container(Container& cont, BinaryPredicate pred)
  79. {
  80. typedef BOOST_DEDUCED_TYPENAME Container::value_type value_t;
  81. std::vector<value_t> temp(cont.begin(), cont.end());
  82. std::sort(temp.begin(), temp.end(), pred);
  83. cont.assign(temp.begin(), temp.end());
  84. }
  85. template<class Container1,
  86. class Container2,
  87. class BinaryPredicate>
  88. void test_pred(Container1 cont1, Container2 cont2, BinaryPredicate pred)
  89. {
  90. typedef BOOST_DEDUCED_TYPENAME Container1::value_type value_t;
  91. typedef BOOST_DEDUCED_TYPENAME std::vector<value_t>::iterator iterator_t;
  92. sort_container(cont1, pred);
  93. sort_container(cont2, pred);
  94. std::vector<value_t> reference_target( cont1.size() + cont2.size() );
  95. iterator_t reference_it
  96. = std::merge(cont1.begin(), cont1.end(),
  97. cont2.begin(), cont2.end(),
  98. reference_target.begin(), pred);
  99. std::vector<value_t> test_target( cont1.size() + cont2.size() );
  100. iterator_t test_it
  101. = boost::merge(cont1, cont2, test_target.begin(), pred);
  102. BOOST_CHECK_EQUAL(
  103. std::distance(reference_target.begin(), reference_it),
  104. std::distance(test_target.begin(), test_it)
  105. );
  106. BOOST_CHECK_EQUAL_COLLECTIONS(
  107. reference_target.begin(), reference_target.end(),
  108. test_target.begin(), test_target.end()
  109. );
  110. test_it = boost::merge(boost::make_iterator_range(cont1), cont2,
  111. test_target.begin(), pred);
  112. BOOST_CHECK_EQUAL(
  113. std::distance(reference_target.begin(), reference_it),
  114. std::distance(test_target.begin(), test_it)
  115. );
  116. BOOST_CHECK_EQUAL_COLLECTIONS(
  117. reference_target.begin(), reference_target.end(),
  118. test_target.begin(), test_target.end()
  119. );
  120. test_it = boost::merge(cont1, boost::make_iterator_range(cont2),
  121. test_target.begin(), pred);
  122. BOOST_CHECK_EQUAL(
  123. std::distance(reference_target.begin(), reference_it),
  124. std::distance(test_target.begin(), test_it)
  125. );
  126. BOOST_CHECK_EQUAL_COLLECTIONS(
  127. reference_target.begin(), reference_target.end(),
  128. test_target.begin(), test_target.end()
  129. );
  130. test_it = boost::merge(boost::make_iterator_range(cont1),
  131. boost::make_iterator_range(cont2),
  132. test_target.begin(), pred);
  133. BOOST_CHECK_EQUAL(
  134. std::distance(reference_target.begin(), reference_it),
  135. std::distance(test_target.begin(), test_it)
  136. );
  137. BOOST_CHECK_EQUAL_COLLECTIONS(
  138. reference_target.begin(), reference_target.end(),
  139. test_target.begin(), test_target.end()
  140. );
  141. }
  142. template<class Container1, class Container2>
  143. void test_merge_impl(Container1& cont1, Container2& cont2)
  144. {
  145. test(cont1, cont2);
  146. test_pred(cont1, cont2, std::less<int>());
  147. test_pred(cont1, cont2, std::greater<int>());
  148. }
  149. template<class Container1, class Container2>
  150. void test_merge_impl()
  151. {
  152. using namespace boost::assign;
  153. Container1 cont1;
  154. Container2 cont2;
  155. test_merge_impl(cont1, cont2);
  156. cont1.clear();
  157. cont2.clear();
  158. cont1 += 1;
  159. test_merge_impl(cont1, cont2);
  160. cont1.clear();
  161. cont2.clear();
  162. cont2 += 1;
  163. test_merge_impl(cont1, cont2);
  164. cont1.clear();
  165. cont2.clear();
  166. cont1 += 1,3,5,7,9,11,13,15,17,19;
  167. cont2 += 2,4,6,8,10,12,14,16,18,20;
  168. test_merge_impl(cont1, cont2);
  169. }
  170. void test_merge()
  171. {
  172. test_merge_impl< std::vector<int>, std::vector<int> >();
  173. test_merge_impl< std::list<int>, std::list<int> >();
  174. test_merge_impl< std::deque<int>, std::deque<int> >();
  175. test_merge_impl< std::list<int>, std::vector<int> >();
  176. test_merge_impl< std::vector<int>, std::list<int> >();
  177. }
  178. }
  179. }
  180. boost::unit_test::test_suite*
  181. init_unit_test_suite(int argc, char* argv[])
  182. {
  183. boost::unit_test::test_suite* test
  184. = BOOST_TEST_SUITE( "RangeTestSuite.algorithm.merge" );
  185. test->add( BOOST_TEST_CASE( &boost::test_merge ) );
  186. return test;
  187. }