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 // <set>
11
12 // class set
13
14 // set& operator=(const set& s);
15
16 #include <set>
17 #include <cassert>
18
19 #include "../../../test_compare.h"
20 #include "../../../test_allocator.h"
21
main()22 int main()
23 {
24 {
25 typedef int V;
26 V ar[] =
27 {
28 1,
29 1,
30 1,
31 2,
32 2,
33 2,
34 3,
35 3,
36 3
37 };
38 typedef test_compare<std::less<int> > C;
39 typedef test_allocator<V> A;
40 std::set<int, C, A> mo(ar, ar+sizeof(ar)/sizeof(ar[0]), C(5), A(2));
41 std::set<int, C, A> m(ar, ar+sizeof(ar)/sizeof(ar[0])/2, C(3), A(7));
42 m = mo;
43 assert(m.get_allocator() == A(7));
44 assert(m.key_comp() == C(5));
45 assert(m.size() == 3);
46 assert(distance(m.begin(), m.end()) == 3);
47 assert(*m.begin() == 1);
48 assert(*next(m.begin()) == 2);
49 assert(*next(m.begin(), 2) == 3);
50
51 assert(mo.get_allocator() == A(2));
52 assert(mo.key_comp() == C(5));
53 assert(mo.size() == 3);
54 assert(distance(mo.begin(), mo.end()) == 3);
55 assert(*mo.begin() == 1);
56 assert(*next(mo.begin()) == 2);
57 assert(*next(mo.begin(), 2) == 3);
58 }
59 {
60 typedef int V;
61 V ar[] =
62 {
63 1,
64 1,
65 1,
66 2,
67 2,
68 2,
69 3,
70 3,
71 3
72 };
73 typedef test_compare<std::less<int> > C;
74 typedef other_allocator<V> A;
75 std::set<int, C, A> mo(ar, ar+sizeof(ar)/sizeof(ar[0]), C(5), A(2));
76 std::set<int, C, A> m(ar, ar+sizeof(ar)/sizeof(ar[0])/2, C(3), A(7));
77 m = mo;
78 assert(m.get_allocator() == A(2));
79 assert(m.key_comp() == C(5));
80 assert(m.size() == 3);
81 assert(distance(m.begin(), m.end()) == 3);
82 assert(*m.begin() == 1);
83 assert(*next(m.begin()) == 2);
84 assert(*next(m.begin(), 2) == 3);
85
86 assert(mo.get_allocator() == A(2));
87 assert(mo.key_comp() == C(5));
88 assert(mo.size() == 3);
89 assert(distance(mo.begin(), mo.end()) == 3);
90 assert(*mo.begin() == 1);
91 assert(*next(mo.begin()) == 2);
92 assert(*next(mo.begin(), 2) == 3);
93 }
94 }
95