xref: /llvm-project/libcxx/test/std/containers/associative/set/contains_transparent.pass.cpp (revision 3fca07d7b9a29fe829c8f04642d35a3bbb94d4d1)
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 // <set>
12 
13 // class set
14 
15 // template<typename K> bool contains(const K& x) const; // C++20
16 
17 #include <cassert>
18 #include <set>
19 #include <utility>
20 
21 struct Comp {
22   using is_transparent = void;
23 
operator ()Comp24   bool operator()(const std::pair<int, int>& lhs,
25                   const std::pair<int, int>& rhs) const {
26     return lhs < rhs;
27   }
28 
operator ()Comp29   bool operator()(const std::pair<int, int>& lhs, int rhs) const {
30     return lhs.first < rhs;
31   }
32 
operator ()Comp33   bool operator()(int lhs, const std::pair<int, int>& rhs) const {
34     return lhs < rhs.first;
35   }
36 };
37 
38 template <typename Container>
test()39 void test() {
40   Container s{{2, 1}, {1, 2}, {1, 3}, {1, 4}, {2, 2}};
41 
42   assert(s.contains(1));
43   assert(!s.contains(-1));
44 }
45 
main(int,char **)46 int main(int, char**) {
47   test<std::set<std::pair<int, int>, Comp> >();
48   test<std::multiset<std::pair<int, int>, Comp> >();
49 
50   return 0;
51 }
52