summaryrefslogtreecommitdiff
path: root/libcxx/test/std/algorithms/alg.modifying.operations/alg.shift/shift_right.pass.cpp
blob: d92387c4c77aeb06aaf0c5c84e3154a15ae2513c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

// UNSUPPORTED: c++03, c++11, c++14, c++17

// <algorithm>

// template<class ForwardIterator>
// constexpr ForwardIterator
//   shift_right(ForwardIterator first, ForwardIterator last,
//               typename iterator_traits<ForwardIterator>::difference_type n);

#include <algorithm>
#include <cassert>

#include "test_macros.h"
#include "test_iterators.h"
#include "MoveOnly.h"

template<class T, class Iter>
constexpr bool test()
{
    int orig[] = {3,1,4,1,5, 9,2,6,5,3, 5,8,9,7,9};
    T work[] = {3,1,4,1,5, 9,2,6,5,3, 5,8,9,7,9};

    for (int n = 0; n <= 15; ++n) {
        for (int k = 0; k <= n+2; ++k) {
            std::copy(orig, orig+n, work);
            Iter it = std::shift_right(Iter(work), Iter(work+n), k);
            if (k < n) {
                assert(it == Iter(work+k));
                assert(std::equal(orig, orig+n-k, work+k, work+n));
            } else {
                assert(it == Iter(work+n));
                assert(std::equal(orig, orig+n, work, work+n));
            }
        }
    }

    // n == 0
    {
        T input[]          = { 0, 1, 2 };
        const T expected[] = { 0, 1, 2 };
        Iter b = Iter(std::begin(input));
        Iter e = Iter(std::end(input));
        Iter it = std::shift_right(b, e, 0);
        assert(std::equal(std::begin(expected), std::end(expected), it, e));
    }

    // n > 0 && n < len
    {
        T input[]          = { 0, 1, 2 };
        const T expected[] = { 0, 1 };
        Iter b = Iter(std::begin(input));
        Iter e = Iter(std::end(input));
        Iter it = std::shift_right(b, e, 1);
        assert(std::equal(std::begin(expected), std::end(expected), it, e));
    }
    {
        T input[]          = { 1, 2, 3, 4, 5, 6, 7, 8 };
        const T expected[] = { 1, 2, 3, 4, 5, 6 };
        Iter b = Iter(std::begin(input));
        Iter e = Iter(std::end(input));
        Iter it = std::shift_right(b, e, 2);
        assert(std::equal(std::begin(expected), std::end(expected), it, e));
    }
    {
        T input[]          = { 1, 2, 3, 4, 5, 6, 7, 8 };
        const T expected[] = { 1, 2 };
        Iter b = Iter(std::begin(input));
        Iter e = Iter(std::end(input));
        Iter it = std::shift_right(b, e, 6);
        assert(std::equal(std::begin(expected), std::end(expected), it, e));
    }

    // n == len
    {
        T input[]          = { 0, 1, 2 };
        const T expected[] = { 0, 1, 2 };
        Iter b = Iter(std::begin(input));
        Iter e = Iter(std::end(input));
        Iter it = std::shift_right(b, e, std::size(input));
        assert(std::equal(std::begin(expected), std::end(expected), b, e));
        assert(it == e);
    }

    // n > len
    {
        T input[]          = { 0, 1, 2 };
        const T expected[] = { 0, 1, 2 };
        Iter b = Iter(std::begin(input));
        Iter e = Iter(std::end(input));
        Iter it = std::shift_right(b, e, std::size(input) + 1);
        assert(std::equal(std::begin(expected), std::end(expected), b, e));
        assert(it == e);
    }

    return true;
}

int main(int, char**)
{
    test<int, forward_iterator<int*>>();
    test<int, bidirectional_iterator<int*>>();
    test<int, random_access_iterator<int*>>();
    test<int, int*>();
    test<MoveOnly, forward_iterator<MoveOnly*>>();
    test<MoveOnly, bidirectional_iterator<MoveOnly*>>();
    test<MoveOnly, random_access_iterator<MoveOnly*>>();
    test<MoveOnly, MoveOnly*>();

    static_assert(test<int, forward_iterator<int*>>());
    static_assert(test<int, bidirectional_iterator<int*>>());
    static_assert(test<int, random_access_iterator<int*>>());
    static_assert(test<int, int*>());
    static_assert(test<MoveOnly, forward_iterator<MoveOnly*>>());
    static_assert(test<MoveOnly, bidirectional_iterator<MoveOnly*>>());
    static_assert(test<MoveOnly, random_access_iterator<MoveOnly*>>());
    static_assert(test<MoveOnly, MoveOnly*>());

    return 0;
}