count_if.cpp 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  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/algorithm/count_if.hpp>
  12. #include <boost/test/test_tools.hpp>
  13. #include <boost/test/unit_test.hpp>
  14. #include <boost/assign.hpp>
  15. #include "../test_function/false_predicate.hpp"
  16. #include "../test_function/true_predicate.hpp"
  17. #include "../test_function/equal_to_x.hpp"
  18. #include <algorithm>
  19. #include <list>
  20. #include <set>
  21. #include <vector>
  22. namespace boost
  23. {
  24. namespace
  25. {
  26. template< class Container >
  27. void test_count_if_impl()
  28. {
  29. using namespace boost::range_test_function;
  30. using namespace boost::assign;
  31. typedef equal_to_x<int> pred_t;
  32. typedef BOOST_DEDUCED_TYPENAME std::iterator_traits<BOOST_DEDUCED_TYPENAME Container::iterator>::difference_type diff_t;
  33. Container cont;
  34. const Container& cref_cont = cont;
  35. BOOST_CHECK_EQUAL( 0u, boost::count_if(cont, pred_t(0)) );
  36. BOOST_CHECK_EQUAL( 0u, boost::count_if(cref_cont, pred_t(0)) );
  37. BOOST_CHECK_EQUAL( 0u, boost::count_if(boost::make_iterator_range(cont), pred_t(0)) );
  38. cont += 1;
  39. BOOST_CHECK_EQUAL( 0u, boost::count_if(cont, pred_t(0)) );
  40. BOOST_CHECK_EQUAL( 0u, boost::count_if(cref_cont, pred_t(0)) );
  41. BOOST_CHECK_EQUAL( 0u, boost::count_if(boost::make_iterator_range(cont), pred_t(0)) );
  42. BOOST_CHECK_EQUAL( 1u, boost::count_if(cont, pred_t(1)) );
  43. BOOST_CHECK_EQUAL( 1u, boost::count_if(cref_cont, pred_t(1)) );
  44. BOOST_CHECK_EQUAL( 1u, boost::count_if(boost::make_iterator_range(cont), pred_t(1)) );
  45. cont += 2,3,4,5,6,7,8,9;
  46. BOOST_CHECK_EQUAL( 0u, boost::count_if(cont, pred_t(0)) );
  47. BOOST_CHECK_EQUAL( 0u, boost::count_if(cref_cont, pred_t(0)) );
  48. BOOST_CHECK_EQUAL( 0u, boost::count_if(boost::make_iterator_range(cont), pred_t(0)) );
  49. BOOST_CHECK_EQUAL( 1u, boost::count_if(cont, pred_t(1)) );
  50. BOOST_CHECK_EQUAL( 1u, boost::count_if(cref_cont, pred_t(1)) );
  51. BOOST_CHECK_EQUAL( 1u, boost::count_if(boost::make_iterator_range(cont), pred_t(1)) );
  52. cont += 2;
  53. BOOST_CHECK_EQUAL( 0u, boost::count_if(cont, pred_t(0)) );
  54. BOOST_CHECK_EQUAL( 0u, boost::count_if(cref_cont, pred_t(0)) );
  55. BOOST_CHECK_EQUAL( 0u, boost::count_if(boost::make_iterator_range(cont), pred_t(0)) );
  56. BOOST_CHECK_EQUAL( 1u, boost::count_if(cont, pred_t(1)) );
  57. BOOST_CHECK_EQUAL( 1u, boost::count_if(cref_cont, pred_t(1)) );
  58. BOOST_CHECK_EQUAL( 1u, boost::count_if(boost::make_iterator_range(cont), pred_t(1)) );
  59. BOOST_CHECK_EQUAL( 2u, boost::count_if(cont, pred_t(2)) );
  60. BOOST_CHECK_EQUAL( 2u, boost::count_if(cref_cont, pred_t(2)) );
  61. BOOST_CHECK_EQUAL( 2u, boost::count_if(boost::make_iterator_range(cont), pred_t(2)) );
  62. BOOST_CHECK_EQUAL( 0u, boost::count_if(cont, false_predicate()) );
  63. BOOST_CHECK_EQUAL( 0u, boost::count_if(cref_cont, false_predicate()) );
  64. BOOST_CHECK_EQUAL( 0u, boost::count_if(boost::make_iterator_range(cont), false_predicate()) );
  65. BOOST_CHECK_EQUAL( static_cast<diff_t>(cont.size()), boost::count_if(cont, true_predicate()) );
  66. BOOST_CHECK_EQUAL( static_cast<diff_t>(cont.size()), boost::count_if(cref_cont, true_predicate()) );
  67. BOOST_CHECK_EQUAL( static_cast<diff_t>(cont.size()), boost::count_if(boost::make_iterator_range(cont), true_predicate()) );
  68. }
  69. void test_count_if()
  70. {
  71. test_count_if_impl< std::vector<int> >();
  72. test_count_if_impl< std::list<int> >();
  73. test_count_if_impl< std::multiset<int> >();
  74. }
  75. }
  76. }
  77. boost::unit_test::test_suite*
  78. init_unit_test_suite(int argc, char* argv[])
  79. {
  80. boost::unit_test::test_suite* test
  81. = BOOST_TEST_SUITE( "RangeTestSuite.algorithm.count_if" );
  82. test->add( BOOST_TEST_CASE( &boost::test_count_if ) );
  83. return test;
  84. }