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 
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