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 // <algorithm>
11
12 // template<BidirectionalIterator Iter, Predicate<auto, Iter::value_type> Pred>
13 // requires ShuffleIterator<Iter>
14 // && CopyConstructible<Pred>
15 // Iter
16 // partition(Iter first, Iter last, Pred pred);
17
18 #include <algorithm>
19 #include <cassert>
20
21
22 #include "test_iterators.h"
23
24 struct is_odd
25 {
operator ()is_odd26 bool operator()(const int& i) const {return i & 1;}
27 };
28
29 template <class Iter>
30 void
test()31 test()
32 {
33 // check mixed
34 int ia[] = {1, 2, 3, 4, 5, 6, 7, 8 ,9};
35 const unsigned sa = sizeof(ia)/sizeof(ia[0]);
36 Iter r = std::partition(Iter(ia), Iter(ia + sa), is_odd());
37 assert(base(r) == ia + 5);
38 for (int* i = ia; i < base(r); ++i)
39 assert(is_odd()(*i));
40 for (int* i = base(r); i < ia+sa; ++i)
41 assert(!is_odd()(*i));
42 // check empty
43 r = std::partition(Iter(ia), Iter(ia), is_odd());
44 assert(base(r) == ia);
45 // check all false
46 for (unsigned i = 0; i < sa; ++i)
47 ia[i] = 2*i;
48 r = std::partition(Iter(ia), Iter(ia+sa), is_odd());
49 assert(base(r) == ia);
50 // check all true
51 for (unsigned i = 0; i < sa; ++i)
52 ia[i] = 2*i+1;
53 r = std::partition(Iter(ia), Iter(ia+sa), is_odd());
54 assert(base(r) == ia+sa);
55 // check all true but last
56 for (unsigned i = 0; i < sa; ++i)
57 ia[i] = 2*i+1;
58 ia[sa-1] = 10;
59 r = std::partition(Iter(ia), Iter(ia+sa), is_odd());
60 assert(base(r) == ia+sa-1);
61 for (int* i = ia; i < base(r); ++i)
62 assert(is_odd()(*i));
63 for (int* i = base(r); i < ia+sa; ++i)
64 assert(!is_odd()(*i));
65 // check all true but first
66 for (unsigned i = 0; i < sa; ++i)
67 ia[i] = 2*i+1;
68 ia[0] = 10;
69 r = std::partition(Iter(ia), Iter(ia+sa), is_odd());
70 assert(base(r) == ia+sa-1);
71 for (int* i = ia; i < base(r); ++i)
72 assert(is_odd()(*i));
73 for (int* i = base(r); i < ia+sa; ++i)
74 assert(!is_odd()(*i));
75 // check all false but last
76 for (unsigned i = 0; i < sa; ++i)
77 ia[i] = 2*i;
78 ia[sa-1] = 11;
79 r = std::partition(Iter(ia), Iter(ia+sa), is_odd());
80 assert(base(r) == ia+1);
81 for (int* i = ia; i < base(r); ++i)
82 assert(is_odd()(*i));
83 for (int* i = base(r); i < ia+sa; ++i)
84 assert(!is_odd()(*i));
85 // check all false but first
86 for (unsigned i = 0; i < sa; ++i)
87 ia[i] = 2*i;
88 ia[0] = 11;
89 r = std::partition(Iter(ia), Iter(ia+sa), is_odd());
90 assert(base(r) == ia+1);
91 for (int* i = ia; i < base(r); ++i)
92 assert(is_odd()(*i));
93 for (int* i = base(r); i < ia+sa; ++i)
94 assert(!is_odd()(*i));
95 }
96
main()97 int main()
98 {
99 test<bidirectional_iterator<int*> >();
100 test<random_access_iterator<int*> >();
101 test<int*>();
102 }
103