drop_while.cpp 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  1. // Range v3 library
  2. //
  3. // Copyright Eric Niebler 2014-present
  4. //
  5. // Use, modification and distribution is subject to the
  6. // Boost Software License, Version 1.0. (See accompanying
  7. // file LICENSE_1_0.txt or copy at
  8. // http://www.boost.org/LICENSE_1_0.txt)
  9. //
  10. // Project home: https://github.com/ericniebler/range-v3
  11. #include <list>
  12. #include <range/v3/core.hpp>
  13. #include <range/v3/view/iota.hpp>
  14. #include <range/v3/view/take.hpp>
  15. #include <range/v3/view/drop_while.hpp>
  16. #include <range/v3/utility/copy.hpp>
  17. #include "../simple_test.hpp"
  18. #include "../test_utils.hpp"
  19. struct my_data
  20. {
  21. int i;
  22. };
  23. bool operator==(my_data left, my_data right)
  24. {
  25. return left.i == right.i;
  26. }
  27. int main()
  28. {
  29. using namespace ranges;
  30. auto rng0 = views::iota(10) | views::drop_while([](int i) { return i < 25; });
  31. CPP_assert(range_cardinality<decltype(rng0)>::value == unknown);
  32. CPP_assert(view_<decltype(rng0)>);
  33. CPP_assert(random_access_range<decltype(rng0)>);
  34. CPP_assert(!common_range<decltype(rng0)>);
  35. CPP_assert(!sized_range<decltype(rng0)>);
  36. CPP_assert(random_access_iterator<decltype(rng0.begin())>);
  37. auto b = rng0.begin();
  38. CHECK(*b == 25);
  39. CHECK(*(b+1) == 26);
  40. ::check_equal(rng0 | views::take(10), {25, 26, 27, 28, 29, 30, 31, 32, 33, 34});
  41. std::list<int> vi{0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
  42. auto rng1 = vi | views::drop_while([](int i) { return i != 50; });
  43. CPP_assert(range_cardinality<decltype(rng1)>::value == ranges::finite);
  44. CPP_assert(view_<decltype(rng1)>);
  45. CPP_assert(bidirectional_range<decltype(rng1)>);
  46. CPP_assert(common_range<decltype(rng1)>);
  47. CPP_assert(!sized_range<decltype(rng1)>);
  48. CPP_assert(bidirectional_iterator<decltype(rng1.begin())>);
  49. CHECK(rng1.begin() == rng1.end());
  50. // Check with a mutable predicate
  51. static int const rgi[] = {0,1,2,3,4,5,6,7,8,9};
  52. int cnt = 0;
  53. auto mutable_only = views::drop_while(rgi, [cnt](int) mutable { return ++cnt <= 5;});
  54. ::check_equal(mutable_only, {5,6,7,8,9});
  55. CPP_assert(view_<decltype(mutable_only)>);
  56. CPP_assert(!view_<decltype(mutable_only) const>);
  57. {
  58. auto rng = debug_input_view<const int>{rgi} | views::drop_while([](int i){ return i < 4; });
  59. using R = decltype(rng);
  60. CPP_assert(input_range<R> && view_<R>);
  61. CPP_assert(!forward_range<R>);
  62. CPP_assert(!common_range<R>);
  63. CPP_assert(!sized_range<R>);
  64. CPP_assert(same_as<int const&, range_reference_t<R>>);
  65. ::check_equal(rng, {4,5,6,7,8,9});
  66. }
  67. {
  68. // with projection
  69. const std::list<my_data> data_list{{1}, {2}, {3}, {1}};
  70. auto rng = data_list | views::drop_while([](int i){ return i <= 2; }, &my_data::i);
  71. ::check_equal(rng, std::list<my_data>{{3}, {1}});
  72. }
  73. auto rng2 = rgi | views::drop_while([](int i) { return i != 50; });
  74. CPP_assert(range_cardinality<decltype(rng2)>::value == ranges::finite);
  75. CPP_assert(view_<decltype(rng2)>);
  76. CPP_assert(contiguous_range<decltype(rng2)>);
  77. CPP_assert(common_range<decltype(rng2)>);
  78. CPP_assert(!sized_range<decltype(rng2)>);
  79. CPP_assert(contiguous_iterator<decltype(rng2.begin())>);
  80. CHECK(ranges::size(rng2) == 0u);
  81. return test_result();
  82. }