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> 24 void test0(S s, Pred p, S expected, 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> 32 void test() 33 { 34 auto isA = [](auto ch) { return ch == 'a';}; 35 auto isB = [](auto ch) { return ch == 'b';}; 36 auto isC = [](auto ch) { return ch == 'c';}; 37 auto isD = [](auto ch) { return ch == 'd';}; 38 auto True = [](auto) { return true; }; 39 auto False = [](auto) { return false; }; 40 41 test0(S(""), isA, S(""), 0); 42 43 test0(S("a"), isA, S(""), 1); 44 test0(S("a"), isB, S("a"), 0); 45 46 test0(S("ab"), isA, S("b"), 1); 47 test0(S("ab"), isB, S("a"), 1); 48 test0(S("ab"), isC, S("ab"), 0); 49 test0(S("aa"), isA, S(""), 2); 50 test0(S("aa"), isC, S("aa"), 0); 51 52 test0(S("abc"), isA, S("bc"), 1); 53 test0(S("abc"), isB, S("ac"), 1); 54 test0(S("abc"), isC, S("ab"), 1); 55 test0(S("abc"), isD, S("abc"), 0); 56 57 test0(S("aab"), isA, S("b"), 2); 58 test0(S("aab"), isB, S("aa"), 1); 59 test0(S("aab"), isC, S("aab"), 0); 60 test0(S("abb"), isA, S("bb"), 1); 61 test0(S("abb"), isB, S("a"), 2); 62 test0(S("abb"), isC, S("abb"), 0); 63 test0(S("aaa"), isA, S(""), 3); 64 test0(S("aaa"), isB, S("aaa"), 0); 65 66 test0(S("aba"), False, S("aba"), 0); 67 test0(S("aba"), True, S(""), 3); 68 } 69 70 int main(int, char**) 71 { 72 test<std::string>(); 73 test<std::basic_string<char, std::char_traits<char>, min_allocator<char>>> (); 74 test<std::basic_string<char, std::char_traits<char>, test_allocator<char>>> (); 75 76 return 0; 77 } 78