libstdc++-v3/testsuite/std/ranges/adaptors/split.cc

   1 // Copyright (C) 2020 Free Software Foundation, Inc.
   2 //
   3 // This file is part of the GNU ISO C++ Library.  This library is free
   4 // software; you can redistribute it and/or modify it under the
   5 // terms of the GNU General Public License as published by the
   6 // Free Software Foundation; either version 3, or (at your option)
   7 // any later version.
   8
   9 // This library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13
  14 // You should have received a copy of the GNU General Public License along
  15 // with this library; see the file COPYING3.  If not see
  16 // <http://www.gnu.org/licenses/>.
  17
  18 // { dg-options "-std=gnu++2a" }
  19 // { dg-do run { target c++2a } }
  20
  21 #include <algorithm>
  22 #include <ranges>
  23 #include <string>
  24 #include <string_view>
  25 #include <testsuite_hooks.h>
  26 #include <testsuite_iterators.h>
  27
  28 using __gnu_test::test_range;
  29 using __gnu_test::forward_iterator_wrapper;
  30
  31 namespace ranges = std::ranges;
  32 namespace views = std::ranges::views;
  33
  34 using namespace std::literals;
  35
  36 void
  37 test01()
  38 {
  39   auto x = "the  quick  brown  fox"sv;
  40   auto p = std::string{"  "};
  41   auto v = x | views::split(p);
  42   auto i = v.begin();
  43   VERIFY( ranges::equal(*i++, "the"sv) );
  44   VERIFY( ranges::equal(*i++, "quick"sv) );
  45   VERIFY( ranges::equal(*i++, "brown"sv) );
  46   VERIFY( ranges::equal(*i++, "fox"sv) );
  47   VERIFY( i == v.end() );
  48 }
  49
  50 void
  51 test02()
  52 {
  53   auto x = "the quick brown fox"sv;
  54   auto v = x | views::split(' ');
  55   auto i = v.begin();
  56   VERIFY( ranges::equal(*i++, "the"sv) );
  57   VERIFY( ranges::equal(*i++, "quick"sv) );
  58   VERIFY( ranges::equal(*i++, "brown"sv) );
  59   VERIFY( ranges::equal(*i++, "fox"sv) );
  60   VERIFY( i == v.end() );
  61 }
  62
  63 void
  64 test03()
  65 {
  66   char x[] = "the quick brown fox";
  67   test_range<char, forward_iterator_wrapper> rx(x, x+sizeof(x)-1);
  68   auto v = rx | views::split(' ');
  69   auto i = v.begin();
  70   VERIFY( ranges::equal(*i++, "the"sv) );
  71   VERIFY( ranges::equal(*i++, "quick"sv) );
  72   VERIFY( ranges::equal(*i++, "brown"sv) );
  73   VERIFY( ranges::equal(*i++, "fox"sv) );
  74   VERIFY( i == v.end() );
  75 }
  76
  77 void
  78 test04()
  79 {
  80   auto x = "the  quick  brown  fox"sv;
  81   std::initializer_list<char> p = {' ', ' '};
  82   static_assert(!ranges::view<decltype(p)>);
  83   static_assert(std::same_as<decltype(p | views::all),
  84                              ranges::ref_view<decltype(p)>>);
  85   auto v = x | views::split(p);
  86   auto i = v.begin();
  87   VERIFY( ranges::equal(*i++, "the"sv) );
  88   VERIFY( ranges::equal(*i++, "quick"sv) );
  89   VERIFY( ranges::equal(*i++, "brown"sv) );
  90   VERIFY( ranges::equal(*i++, "fox"sv) );
  91   VERIFY( i == v.end() );
  92 }
  93
  94 void
  95 test05()
  96 {
  97   auto as_string = [](ranges::view auto rng) {
  98     auto in = rng | views::common;
  99     return std::string(in.begin(), in.end());
 100   };
 101   std::string str
 102     = "Now is the time for all good men to come to the aid of their county.";
 103   auto rng
 104     = str | views::split(' ') | views::transform(as_string) | views::common;
 105   std::vector<std::string> words(rng.begin(), rng.end());
 106   auto not_space_p = [](char c) { return c != ' '; };
 107   VERIFY( ranges::equal(words | views::join,
 108                         str | views::filter(not_space_p)) );
 109 }
 110
 111 int
 112 main()
 113 {
 114   test01();
 115   test02();
 116   test03();
 117   test04();
 118   test05();
 119 }