erase_key.pass.cpp revision 061d0cc4db18d17bf01ed14c5db0be098205bd47 
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
10// <unordered_map>
11
12// template <class Key, class T, class Hash = hash<Key>, class Pred = equal_to<Key>,
13//           class Alloc = allocator<pair<const Key, T>>>
14// class unordered_map
15
16// size_type erase(const key_type& k);
17
18#include <unordered_map>
19#include <string>
20#include <cassert>
21
22#include "min_allocator.h"
23
24int main()
25{
26    {
27        typedef std::unordered_map<int, std::string> C;
28        typedef std::pair<int, std::string> P;
29        P a[] =
30        {
31            P(1, "one"),
32            P(2, "two"),
33            P(3, "three"),
34            P(4, "four"),
35            P(1, "four"),
36            P(2, "four"),
37        };
38        C c(a, a + sizeof(a)/sizeof(a[0]));
39        assert(c.erase(5) == 0);
40        assert(c.size() == 4);
41        assert(c.at(1) == "one");
42        assert(c.at(2) == "two");
43        assert(c.at(3) == "three");
44        assert(c.at(4) == "four");
45
46        assert(c.erase(2) == 1);
47        assert(c.size() == 3);
48        assert(c.at(1) == "one");
49        assert(c.at(3) == "three");
50        assert(c.at(4) == "four");
51
52        assert(c.erase(2) == 0);
53        assert(c.size() == 3);
54        assert(c.at(1) == "one");
55        assert(c.at(3) == "three");
56        assert(c.at(4) == "four");
57
58        assert(c.erase(4) == 1);
59        assert(c.size() == 2);
60        assert(c.at(1) == "one");
61        assert(c.at(3) == "three");
62
63        assert(c.erase(4) == 0);
64        assert(c.size() == 2);
65        assert(c.at(1) == "one");
66        assert(c.at(3) == "three");
67
68        assert(c.erase(1) == 1);
69        assert(c.size() == 1);
70        assert(c.at(3) == "three");
71
72        assert(c.erase(1) == 0);
73        assert(c.size() == 1);
74        assert(c.at(3) == "three");
75
76        assert(c.erase(3) == 1);
77        assert(c.size() == 0);
78
79        assert(c.erase(3) == 0);
80        assert(c.size() == 0);
81    }
82#if __cplusplus >= 201103L
83    {
84        typedef std::unordered_map<int, std::string, std::hash<int>, std::equal_to<int>,
85                            min_allocator<std::pair<const int, std::string>>> C;
86        typedef std::pair<int, std::string> P;
87        P a[] =
88        {
89            P(1, "one"),
90            P(2, "two"),
91            P(3, "three"),
92            P(4, "four"),
93            P(1, "four"),
94            P(2, "four"),
95        };
96        C c(a, a + sizeof(a)/sizeof(a[0]));
97        assert(c.erase(5) == 0);
98        assert(c.size() == 4);
99        assert(c.at(1) == "one");
100        assert(c.at(2) == "two");
101        assert(c.at(3) == "three");
102        assert(c.at(4) == "four");
103
104        assert(c.erase(2) == 1);
105        assert(c.size() == 3);
106        assert(c.at(1) == "one");
107        assert(c.at(3) == "three");
108        assert(c.at(4) == "four");
109
110        assert(c.erase(2) == 0);
111        assert(c.size() == 3);
112        assert(c.at(1) == "one");
113        assert(c.at(3) == "three");
114        assert(c.at(4) == "four");
115
116        assert(c.erase(4) == 1);
117        assert(c.size() == 2);
118        assert(c.at(1) == "one");
119        assert(c.at(3) == "three");
120
121        assert(c.erase(4) == 0);
122        assert(c.size() == 2);
123        assert(c.at(1) == "one");
124        assert(c.at(3) == "three");
125
126        assert(c.erase(1) == 1);
127        assert(c.size() == 1);
128        assert(c.at(3) == "three");
129
130        assert(c.erase(1) == 0);
131        assert(c.size() == 1);
132        assert(c.at(3) == "three");
133
134        assert(c.erase(3) == 1);
135        assert(c.size() == 0);
136
137        assert(c.erase(3) == 0);
138        assert(c.size() == 0);
139    }
140#endif
141}
142