libcxx/test/std/containers/associative/set/contains_transparent.pass.cpp

   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
  24   bool operator()(const std::pair<int, int>& lhs,
  25                   const std::pair<int, int>& rhs) const {
  26     return lhs < rhs;
  27   }
  28
  29   bool operator()(const std::pair<int, int>& lhs, int rhs) const {
  30     return lhs.first < rhs;
  31   }
  32
  33   bool operator()(int lhs, const std::pair<int, int>& rhs) const {
  34     return lhs < rhs.first;
  35   }
  36 };
  37
  38 template <typename Container>
  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
  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 }