1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 // UNSUPPORTED: c++03, c++11, c++14, c++17
10
11 // <string>
12
13 // template <class charT, class traits, class Allocator, class Predicate>
14 // typename basic_string<charT, traits, Allocator>::size_type
15 // erase_if(basic_string<charT, traits, Allocator>& c, Predicate pred);
16
17 #include <string>
18
19 #include "test_macros.h"
20 #include "test_allocator.h"
21 #include "min_allocator.h"
22
23 template <class S, class Pred>
test0(S s,Pred p,S expected,std::size_t expected_erased_count)24 void test0(S s, Pred p, S expected, std::size_t expected_erased_count) {
25 ASSERT_SAME_TYPE(typename S::size_type, decltype(std::erase_if(s, p)));
26 assert(expected_erased_count == std::erase_if(s, p));
27 LIBCPP_ASSERT(s.__invariants());
28 assert(s == expected);
29 }
30
31 template <typename S>
test()32 void test() {
33 auto isA = [](auto ch) { return ch == 'a'; };
34 auto isB = [](auto ch) { return ch == 'b'; };
35 auto isC = [](auto ch) { return ch == 'c'; };
36 auto isD = [](auto ch) { return ch == 'd'; };
37 auto True = [](auto) { return true; };
38 auto False = [](auto) { return false; };
39
40 test0(S(""), isA, S(""), 0);
41
42 test0(S("a"), isA, S(""), 1);
43 test0(S("a"), isB, S("a"), 0);
44
45 test0(S("ab"), isA, S("b"), 1);
46 test0(S("ab"), isB, S("a"), 1);
47 test0(S("ab"), isC, S("ab"), 0);
48 test0(S("aa"), isA, S(""), 2);
49 test0(S("aa"), isC, S("aa"), 0);
50
51 test0(S("abc"), isA, S("bc"), 1);
52 test0(S("abc"), isB, S("ac"), 1);
53 test0(S("abc"), isC, S("ab"), 1);
54 test0(S("abc"), isD, S("abc"), 0);
55
56 test0(S("aab"), isA, S("b"), 2);
57 test0(S("aab"), isB, S("aa"), 1);
58 test0(S("aab"), isC, S("aab"), 0);
59 test0(S("abb"), isA, S("bb"), 1);
60 test0(S("abb"), isB, S("a"), 2);
61 test0(S("abb"), isC, S("abb"), 0);
62 test0(S("aaa"), isA, S(""), 3);
63 test0(S("aaa"), isB, S("aaa"), 0);
64
65 test0(S("aba"), False, S("aba"), 0);
66 test0(S("aba"), True, S(""), 3);
67 }
68
main(int,char **)69 int main(int, char**) {
70 test<std::string>();
71 test<std::basic_string<char, std::char_traits<char>, min_allocator<char>>>();
72 test<std::basic_string<char, std::char_traits<char>, test_allocator<char>>>();
73
74 return 0;
75 }
76