xref: /llvm-project/libcxx/test/std/containers/associative/map/map.ops/equal_range_transparent.pass.cpp (revision cd854e686f0c3bfc8bcf4d6fc2549115010da260)
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
10 
11 // <map>
12 
13 // class map
14 
15 // template<typename K>
16 //         pair<iterator,iterator>             equal_range(const K& x); // C++14
17 // template<typename K>
18 //         pair<const_iterator,const_iterator> equal_range(const K& x) const;
19 //         // C++14
20 
21 #include <cassert>
22 #include <map>
23 #include <utility>
24 
25 struct Comp {
26   using is_transparent = void;
27 
operator ()Comp28   bool operator()(const std::pair<int, int> &lhs,
29                   const std::pair<int, int> &rhs) const {
30     return lhs < rhs;
31   }
32 
operator ()Comp33   bool operator()(const std::pair<int, int> &lhs, int rhs) const {
34     return lhs.first < rhs;
35   }
36 
operator ()Comp37   bool operator()(int lhs, const std::pair<int, int> &rhs) const {
38     return lhs < rhs.first;
39   }
40 };
41 
main(int,char **)42 int main(int, char**) {
43   std::map<std::pair<int, int>, int, Comp> s{
44       {{2, 1}, 1}, {{1, 2}, 2}, {{1, 3}, 3}, {{1, 4}, 4}, {{2, 2}, 5}};
45 
46   auto er = s.equal_range(1);
47   long nels = 0;
48 
49   for (auto it = er.first; it != er.second; it++) {
50     assert(it->first.first == 1);
51     nels++;
52   }
53 
54   assert(nels == 3);
55 
56   return 0;
57 }
58