1 //===----------------------------------------------------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 // <valarray> 10 11 // template<class T> class valarray; 12 13 // valarray shift(int i) const; 14 15 #include <valarray> 16 #include <cassert> 17 18 #include "test_macros.h" 19 20 int main(int, char**) 21 { 22 { 23 typedef int T; 24 T a1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 25 T a2[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 26 const unsigned N1 = sizeof(a1)/sizeof(a1[0]); 27 std::valarray<T> v1(a1, N1); 28 std::valarray<T> v2 = v1.shift(0); 29 assert(v2.size() == N1); 30 for (unsigned i = 0; i < N1; ++i) 31 assert(v2[i] == a2[i]); 32 } 33 { 34 typedef int T; 35 T a1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 36 T a2[] = {2, 3, 4, 5, 6, 7, 8, 9, 10, 0}; 37 const unsigned N1 = sizeof(a1)/sizeof(a1[0]); 38 std::valarray<T> v1(a1, N1); 39 std::valarray<T> v2 = v1.shift(1); 40 assert(v2.size() == N1); 41 for (unsigned i = 0; i < N1; ++i) 42 assert(v2[i] == a2[i]); 43 } 44 { 45 typedef int T; 46 T a1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 47 T a2[] = {10, 0, 0, 0, 0, 0, 0, 0, 0, 0}; 48 const unsigned N1 = sizeof(a1)/sizeof(a1[0]); 49 std::valarray<T> v1(a1, N1); 50 std::valarray<T> v2 = v1.shift(9); 51 assert(v2.size() == N1); 52 for (unsigned i = 0; i < N1; ++i) 53 assert(v2[i] == a2[i]); 54 } 55 { 56 typedef int T; 57 T a1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 58 T a2[] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0}; 59 const unsigned N1 = sizeof(a1)/sizeof(a1[0]); 60 std::valarray<T> v1(a1, N1); 61 std::valarray<T> v2 = v1.shift(90); 62 assert(v2.size() == N1); 63 for (unsigned i = 0; i < N1; ++i) 64 assert(v2[i] == a2[i]); 65 } 66 { 67 typedef int T; 68 T a1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 69 T a2[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}; 70 const unsigned N1 = sizeof(a1)/sizeof(a1[0]); 71 std::valarray<T> v1(a1, N1); 72 std::valarray<T> v2 = v1.shift(-1); 73 assert(v2.size() == N1); 74 for (unsigned i = 0; i < N1; ++i) 75 assert(v2[i] == a2[i]); 76 } 77 { 78 typedef int T; 79 T a1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 80 T a2[] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 1}; 81 const unsigned N1 = sizeof(a1)/sizeof(a1[0]); 82 std::valarray<T> v1(a1, N1); 83 std::valarray<T> v2 = v1.shift(-9); 84 assert(v2.size() == N1); 85 for (unsigned i = 0; i < N1; ++i) 86 assert(v2[i] == a2[i]); 87 } 88 { 89 typedef int T; 90 T a1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 91 T a2[] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0}; 92 const unsigned N1 = sizeof(a1)/sizeof(a1[0]); 93 std::valarray<T> v1(a1, N1); 94 std::valarray<T> v2 = v1.shift(-90); 95 assert(v2.size() == N1); 96 for (unsigned i = 0; i < N1; ++i) 97 assert(v2[i] == a2[i]); 98 } 99 { 100 typedef int T; 101 const unsigned N1 = 0; 102 std::valarray<T> v1; 103 std::valarray<T> v2 = v1.shift(-90); 104 assert(v2.size() == N1); 105 } 106 { 107 typedef int T; 108 T a1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 109 T a2[] = {8, 10, 12, 14, 16, 18, 20, 0, 0, 0}; 110 const unsigned N1 = sizeof(a1)/sizeof(a1[0]); 111 std::valarray<T> v1(a1, N1); 112 std::valarray<T> v2 = (v1 + v1).shift(3); 113 assert(v2.size() == N1); 114 for (unsigned i = 0; i < N1; ++i) 115 assert(v2[i] == a2[i]); 116 } 117 { 118 typedef int T; 119 T a1[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; 120 T a2[] = {0, 0, 0, 2, 4, 6, 8, 10, 12, 14}; 121 const unsigned N1 = sizeof(a1)/sizeof(a1[0]); 122 std::valarray<T> v1(a1, N1); 123 std::valarray<T> v2 = (v1 + v1).shift(-3); 124 assert(v2.size() == N1); 125 for (unsigned i = 0; i < N1; ++i) 126 assert(v2[i] == a2[i]); 127 } 128 129 return 0; 130 } 131