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 <class InputIterator, class OutputIterator1,
13 // class OutputIterator2, class Predicate>
14 // constexpr pair<OutputIterator1, OutputIterator2> // constexpr after C++17
15 // partition_copy(InputIterator first, InputIterator last,
16 // OutputIterator1 out_true, OutputIterator2 out_false,
17 // Predicate pred);
18
19 #include <algorithm>
20 #include <cassert>
21
22 #include "test_macros.h"
23 #include "test_iterators.h"
24
25 struct is_odd
26 {
operator ()is_odd27 TEST_CONSTEXPR bool operator()(const int& i) const {return i & 1;}
28 };
29
30 #if TEST_STD_VER > 17
test_constexpr()31 TEST_CONSTEXPR bool test_constexpr() {
32 int ia[] = {1, 3, 5, 2, 4, 6};
33 int r1[10] = {0};
34 int r2[10] = {0};
35
36 auto p = std::partition_copy(std::begin(ia), std::end(ia),
37 std::begin(r1), std::begin(r2), is_odd());
38
39 return std::all_of(std::begin(r1), p.first, is_odd())
40 && std::all_of(p.first, std::end(r1), [](int a){return a == 0;})
41 && std::none_of(std::begin(r2), p.second, is_odd())
42 && std::all_of(p.second, std::end(r2), [](int a){return a == 0;})
43 ;
44 }
45 #endif
46
main()47 int main()
48 {
49 {
50 const int ia[] = {1, 2, 3, 4, 6, 8, 5, 7};
51 int r1[10] = {0};
52 int r2[10] = {0};
53 typedef std::pair<output_iterator<int*>, int*> P;
54 P p = std::partition_copy(input_iterator<const int*>(std::begin(ia)),
55 input_iterator<const int*>(std::end(ia)),
56 output_iterator<int*>(r1), r2, is_odd());
57 assert(p.first.base() == r1 + 4);
58 assert(r1[0] == 1);
59 assert(r1[1] == 3);
60 assert(r1[2] == 5);
61 assert(r1[3] == 7);
62 assert(p.second == r2 + 4);
63 assert(r2[0] == 2);
64 assert(r2[1] == 4);
65 assert(r2[2] == 6);
66 assert(r2[3] == 8);
67 }
68
69 #if TEST_STD_VER > 17
70 static_assert(test_constexpr());
71 #endif
72 }
73