1 //===----------------------------------------------------------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is dual licensed under the MIT and the University of Illinois Open 6 // Source Licenses. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // UNSUPPORTED: c++98, c++03, c++11, c++14, c++17 10 11 // <vector> 12 13 // template <class T, class Allocator, class U> 14 // void erase(vector<T, Allocator>& c, const U& value); 15 16 17 #include <vector> 18 #include <optional> 19 20 #include "test_macros.h" 21 #include "test_allocator.h" 22 #include "min_allocator.h" 23 24 template <class S, class U> 25 void 26 test0(S s, U val, S expected) 27 { 28 ASSERT_SAME_TYPE(void, decltype(std::erase(s, val))); 29 std::erase(s, val); 30 assert(s == expected); 31 } 32 33 template <class S> 34 void test() 35 { 36 37 test0(S(), 1, S()); 38 39 test0(S({1}), 1, S()); 40 test0(S({1}), 2, S({1})); 41 42 test0(S({1,2}), 1, S({2})); 43 test0(S({1,2}), 2, S({1})); 44 test0(S({1,2}), 3, S({1,2})); 45 test0(S({1,1}), 1, S()); 46 test0(S({1,1}), 3, S({1,1})); 47 48 test0(S({1,2,3}), 1, S({2,3})); 49 test0(S({1,2,3}), 2, S({1,3})); 50 test0(S({1,2,3}), 3, S({1,2})); 51 test0(S({1,2,3}), 4, S({1,2,3})); 52 53 test0(S({1,1,1}), 1, S()); 54 test0(S({1,1,1}), 2, S({1,1,1})); 55 test0(S({1,1,2}), 1, S({2})); 56 test0(S({1,1,2}), 2, S({1,1})); 57 test0(S({1,1,2}), 3, S({1,1,2})); 58 test0(S({1,2,2}), 1, S({2,2})); 59 test0(S({1,2,2}), 2, S({1})); 60 test0(S({1,2,2}), 3, S({1,2,2})); 61 62 // Test cross-type erasure 63 using opt = std::optional<typename S::value_type>; 64 test0(S({1,2,1}), opt(), S({1,2,1})); 65 test0(S({1,2,1}), opt(1), S({2})); 66 test0(S({1,2,1}), opt(2), S({1,1})); 67 test0(S({1,2,1}), opt(3), S({1,2,1})); 68 } 69 70 int main() 71 { 72 test<std::vector<int>>(); 73 test<std::vector<int, min_allocator<int>>> (); 74 test<std::vector<int, test_allocator<int>>> (); 75 76 test<std::vector<long>>(); 77 test<std::vector<double>>(); 78 } 79
