drop_last.cpp 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213
  1. /// \file
  2. // Range v3 library
  3. //
  4. // Copyright Andrey Diduh 2019
  5. //
  6. // Use, modification and distribution is subject to the
  7. // Boost Software License, Version 1.0. (See accompanying
  8. // file LICENSE_1_0.txt or copy at
  9. // http://www.boost.org/LICENSE_1_0.txt)
  10. //
  11. // Project home: https://github.com/ericniebler/range-v3
  12. //
  13. #include "../simple_test.hpp"
  14. #include "../test_utils.hpp"
  15. #include <type_traits>
  16. #include <vector>
  17. #include <list>
  18. #include <forward_list>
  19. #include <range/v3/view/drop_last.hpp>
  20. #include <range/v3/view/take_exactly.hpp>
  21. #include <range/v3/view/transform.hpp>
  22. #include <range/v3/view/generate_n.hpp>
  23. using namespace ranges;
  24. template<class Rng>
  25. struct view_non_const_only
  26. : view_adaptor<view_non_const_only<Rng>, Rng>
  27. {
  28. private:
  29. friend range_access;
  30. adaptor_base begin_adaptor() { return {}; }
  31. adaptor_base end_adaptor() { return {}; }
  32. public:
  33. using view_non_const_only::view_adaptor::view_adaptor;
  34. CPP_auto_member
  35. auto CPP_fun(size)() (
  36. requires sized_range<Rng>)
  37. {
  38. return ranges::size(this->base());
  39. }
  40. };
  41. template<class Rng>
  42. view_non_const_only<views::all_t<Rng>> non_const_only(Rng &&rng)
  43. {
  44. return view_non_const_only<views::all_t<Rng>>{views::all(static_cast<Rng&&>(rng))};
  45. }
  46. template<class Rng>
  47. void test_range(Rng&& src)
  48. {
  49. // additional src copy for InputStream
  50. {
  51. auto src_ = src;
  52. ::check_equal(src_, {1,2,3,4});
  53. }
  54. {
  55. auto src_ = src;
  56. auto list = src_ | views::drop_last(2);
  57. ::check_equal(list, {1,2});
  58. }
  59. {
  60. auto src_ = src;
  61. auto list = src_ | views::drop_last(0);
  62. ::check_equal(list, {1,2,3,4});
  63. }
  64. {
  65. auto src_ = src;
  66. auto list = src_ | views::drop_last(4);
  67. CHECK(list.empty());
  68. }
  69. {
  70. auto src_ = src;
  71. auto list = src_ | views::drop_last(5);
  72. CHECK(list.empty());
  73. }
  74. }
  75. template<class Rng>
  76. void test_size(Rng&& src)
  77. {
  78. CHECK( (src | views::drop_last(0)).size() == std::size_t(4) );
  79. CHECK( (src | views::drop_last(2)).size() == std::size_t(2) );
  80. CHECK( (src | views::drop_last(4)).size() == std::size_t(0) );
  81. CHECK( (src | views::drop_last(5)).size() == std::size_t(0) );
  82. }
  83. template<class Rng>
  84. void test_non_convert_range(Rng&& src)
  85. {
  86. // test non-convertible const<=>non-const range
  87. test_range(src | views::transform([](const int& i) -> const int& {return i;}));
  88. }
  89. void random_acccess_test()
  90. {
  91. using Src = std::vector<int>;
  92. static_assert(
  93. ranges::random_access_range<Src>
  94. , "Must be exactly RA.");
  95. static_assert(
  96. std::is_same<
  97. drop_last_view<views::all_t<Src&>>,
  98. drop_last_view<views::all_t<Src&>, detail::drop_last_view::mode_bidi>
  99. >::value
  100. , "Must have correct view.");
  101. Src src = {1,2,3,4};
  102. test_range(src);
  103. test_range(non_const_only(src));
  104. test_size(src);
  105. test_non_convert_range(src);
  106. }
  107. void bidirectional_test()
  108. {
  109. using Src = std::list<int>;
  110. static_assert(
  111. !ranges::random_access_range<Src> &&
  112. ranges::bidirectional_range<Src>
  113. , "Must be exactly bidirectional.");
  114. static_assert(
  115. std::is_same<
  116. /* mode_sized for max_performance profile.
  117. * mode_bidi for compatible profile.
  118. * See aux::drop_last::get_mode */
  119. drop_last_view<views::all_t<Src&>>,
  120. drop_last_view<views::all_t<Src&>, detail::drop_last_view::mode_bidi>
  121. >::value
  122. , "Must have correct view.");
  123. Src src = {1,2,3,4};
  124. test_range(src);
  125. test_range(non_const_only(src));
  126. test_size(src);
  127. test_non_convert_range(src);
  128. }
  129. void forward_test()
  130. {
  131. using Src = std::forward_list<int>;
  132. static_assert(
  133. !ranges::bidirectional_range<Src> &&
  134. ranges::forward_range<Src>
  135. , "Must be exactly forward.");
  136. static_assert(
  137. std::is_same<
  138. drop_last_view<views::all_t<Src&>>,
  139. drop_last_view<views::all_t<Src&>, detail::drop_last_view::mode_forward>
  140. >::value
  141. , "Must have correct view.");
  142. Src src = {1,2,3,4};
  143. test_range(src);
  144. test_range(non_const_only(src));
  145. test_size(src | views::take_exactly(4));
  146. test_non_convert_range(src);
  147. }
  148. void sized_test()
  149. {
  150. int i = 0;
  151. auto src = views::generate_n([i]() mutable -> int { return ++i;}, 4);
  152. using Src = decltype(src);
  153. static_assert(
  154. !ranges::forward_range<Src> &&
  155. ranges::input_range<Src>
  156. , "Must be exactly input.");
  157. static_assert(
  158. std::is_same<
  159. drop_last_view<views::all_t<Src>>,
  160. drop_last_view<views::all_t<Src>, detail::drop_last_view::mode_sized>
  161. >::value
  162. , "Must have correct view.");
  163. {
  164. // always non-const
  165. auto src_ = src;
  166. test_range(src_);
  167. }
  168. {
  169. auto src_ = src;
  170. test_size(src_);
  171. }
  172. {
  173. auto src_ = src;
  174. test_range(non_const_only(std::move(src_)));
  175. }
  176. {
  177. auto src_ = src;
  178. test_non_convert_range(src_);
  179. }
  180. }
  181. int main()
  182. {
  183. random_acccess_test();
  184. bidirectional_test();
  185. forward_test();
  186. sized_test();
  187. return test_result();
  188. }