llvm-project

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
// <map>
10

11
// class map
12

13
// size_type count(const key_type& k) const;
14

15
#include <map>
16
#include <cassert>
17

18
#include "test_macros.h"
19
#include "min_allocator.h"
20
#include "private_constructor.h"
21
#include "is_transparent.h"
22

23
#if TEST_STD_VER >= 11
24
template <class T>
25
struct FinalCompare final {
26
  bool operator()(const T& x, const T& y) const { return x < y; }
27
};
28
#endif
29

30
template <class Map, class ArgType = typename Map::key_type>
31
void test() {
32
  typedef typename Map::value_type V;
33
  typedef typename Map::size_type R;
34

35
  V ar[] = {V(5, 5), V(6, 6), V(7, 7), V(8, 8), V(9, 9), V(10, 10), V(11, 11), V(12, 12)};
36

37
  const Map m(ar, ar + sizeof(ar) / sizeof(ar[0]));
38

39
  for (int i = 0; i < 5; ++i) {
40
    R r = m.count(ArgType(i));
41
    assert(r == 0);
42
  }
43

44
  for (int i = 5; i < 13; ++i) {
45
    R r = m.count(ArgType(i));
46
    assert(r == 1);
47
  }
48
}
49

50
int main(int, char**) {
51
  test<std::map<int, double> >();
52
#if TEST_STD_VER >= 11
53
  typedef std::pair<const int, double> V;
54
  test<std::map<int, double, std::less<int>, min_allocator<V>>>();
55
  test<std::map<int, double, FinalCompare<int>>>();
56
#endif
57
#if TEST_STD_VER >= 14
58
  typedef std::map<int, double, std::less<>> TM;
59
  test<TM>();
60
  test<TM, C2Int>();
61

62
  {
63
    typedef PrivateConstructor PC;
64
    typedef std::map<PC, double, std::less<> > M;
65
    typedef M::size_type R;
66

67
    M m;
68
    m[PC::make(5)]  = 5;
69
    m[PC::make(6)]  = 6;
70
    m[PC::make(7)]  = 7;
71
    m[PC::make(8)]  = 8;
72
    m[PC::make(9)]  = 9;
73
    m[PC::make(10)] = 10;
74
    m[PC::make(11)] = 11;
75
    m[PC::make(12)] = 12;
76

77
    for (int i = 0; i < 5; ++i) {
78
      R r = m.count(i);
79
      assert(r == 0);
80
    }
81

82
    for (int i = 5; i < 13; ++i) {
83
      R r = m.count(i);
84
      assert(r == 1);
85
    }
86
  }
87
#endif // TEST_STD_VER >= 14
88
  return 0;
89
}
90