split.cpp 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  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. #include <cctype>
  10. #include <string>
  11. #include <vector>
  12. #include <range/v3/action/split.hpp>
  13. #include <range/v3/action/split_when.hpp>
  14. #include <range/v3/algorithm/move.hpp>
  15. #include <range/v3/core.hpp>
  16. #include <range/v3/view/c_str.hpp>
  17. #include <range/v3/view/iota.hpp>
  18. #include "../simple_test.hpp"
  19. #include "../test_utils.hpp"
  20. int main()
  21. {
  22. using namespace ranges;
  23. {
  24. auto v = views::ints(1, 21) | to<std::vector>();
  25. std::vector<std::vector<int>> rgv = actions::split(v, 10);
  26. CHECK(rgv.size() == 2u);
  27. ::check_equal(rgv[0], {1,2,3,4,5,6,7,8,9});
  28. ::check_equal(rgv[1], {11,12,13,14,15,16,17,18,19,20});
  29. using I = std::vector<int>::iterator;
  30. std::vector<std::vector<int>> rgv2 = actions::split_when(
  31. v, [](I b, I) { return std::make_pair(0 == (*b) % 2, next(b)); });
  32. CHECK(rgv2.size() == 10u);
  33. ::check_equal(rgv2[0], {1});
  34. ::check_equal(rgv2[1], {3});
  35. ::check_equal(rgv2[2], {5});
  36. ::check_equal(rgv2[3], {7});
  37. ::check_equal(rgv2[4], {9});
  38. ::check_equal(rgv2[5], {11});
  39. ::check_equal(rgv2[6], {13});
  40. ::check_equal(rgv2[7], {15});
  41. ::check_equal(rgv2[8], {17});
  42. ::check_equal(rgv2[9], {19});
  43. }
  44. {
  45. std::string s{"This is his face"};
  46. std::vector<std::string> rgs = actions::split(s, views::c_str(" "));
  47. CHECK(rgs.size() == 4u);
  48. CHECK(rgs[0] == "This");
  49. CHECK(rgs[1] == "is");
  50. CHECK(rgs[2] == "his");
  51. CHECK(rgs[3] == "face");
  52. }
  53. {
  54. std::string s{"This is his face"};
  55. std::vector<std::string> rgs = std::move(s) | actions::split(views::c_str(" "));
  56. CHECK(rgs.size() == 4u);
  57. CHECK(rgs[0] == "This");
  58. CHECK(rgs[1] == "is");
  59. CHECK(rgs[2] == "his");
  60. CHECK(rgs[3] == "face");
  61. }
  62. {
  63. std::string s{"This is his face"};
  64. char ch[] = {' '};
  65. std::vector<std::string> rgs = actions::split(s, ch);
  66. CHECK(rgs.size() == 4u);
  67. CHECK(rgs[0] == "This");
  68. CHECK(rgs[1] == "is");
  69. CHECK(rgs[2] == "his");
  70. CHECK(rgs[3] == "face");
  71. }
  72. {
  73. std::string s{"This is his face"};
  74. char ch[] = {' '};
  75. std::vector<std::string> rgs = std::move(s) | actions::split(ch);
  76. CHECK(rgs.size() == 4u);
  77. CHECK(rgs[0] == "This");
  78. CHECK(rgs[1] == "is");
  79. CHECK(rgs[2] == "his");
  80. CHECK(rgs[3] == "face");
  81. }
  82. {
  83. auto rgi = views::ints(1,21);
  84. std::vector<std::vector<int>> rgv3 = actions::split(rgi, 10);
  85. CHECK(rgv3.size() == 2u);
  86. ::check_equal(rgv3[0], {1,2,3,4,5,6,7,8,9});
  87. ::check_equal(rgv3[1], {11,12,13,14,15,16,17,18,19,20});
  88. }
  89. {
  90. auto rgi = views::ints(1,21);
  91. std::vector<std::vector<int>> rgv3 = std::move(rgi) | actions::split(10);
  92. CHECK(rgv3.size() == 2u);
  93. ::check_equal(rgv3[0], {1,2,3,4,5,6,7,8,9});
  94. ::check_equal(rgv3[1], {11,12,13,14,15,16,17,18,19,20});
  95. }
  96. {
  97. std::string str("now is \t the\ttime");
  98. auto toks = actions::split_when(str, +[](int i) { return std::isspace(i); });
  99. static_assert(std::is_same<decltype(toks), std::vector<std::string>>::value, "");
  100. CHECK(toks.size() == 4u);
  101. if(toks.size() == 4u)
  102. {
  103. CHECK(toks[0] == "now");
  104. CHECK(toks[1] == "is");
  105. CHECK(toks[2] == "the");
  106. CHECK(toks[3] == "time");
  107. }
  108. }
  109. {
  110. std::string str("now is \t the\ttime");
  111. auto toks =
  112. std::move(str) | actions::split_when(+[](int i) { return std::isspace(i); });
  113. static_assert(std::is_same<decltype(toks), std::vector<std::string>>::value, "");
  114. CHECK(toks.size() == 4u);
  115. if(toks.size() == 4u)
  116. {
  117. CHECK(toks[0] == "now");
  118. CHECK(toks[1] == "is");
  119. CHECK(toks[2] == "the");
  120. CHECK(toks[3] == "time");
  121. }
  122. }
  123. return ::test_result();
  124. }