81 lines · cpp
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-exception6//7//===----------------------------------------------------------------------===//8 9// <valarray>10 11// template <class T> class gslice_array12 13// void operator>>= (const valarray<value_type>& v) const;14 15#include <valarray>16#include <cassert>17 18#include "test_macros.h"19 20int main(int, char**)21{22 int a1[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11,23 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23,24 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35,25 36, 37, 38, 39, 40};26 int a2[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12,27 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24};28 std::valarray<int> v1(a1, sizeof(a1)/sizeof(a1[0]));29 std::valarray<int> v2(a2, sizeof(a2)/sizeof(a2[0]));30 std::size_t sz[] = {2, 4, 3};31 std::size_t st[] = {19, 4, 1};32 typedef std::valarray<std::size_t> sizes;33 typedef std::valarray<std::size_t> strides;34 v1[std::gslice(3, sizes(sz, sizeof(sz)/sizeof(sz[0])),35 strides(st, sizeof(st)/sizeof(st[0])))] >>= v2;36 assert(v1.size() == 41);37 assert(v1[ 0] == 0);38 assert(v1[ 1] == 1);39 assert(v1[ 2] == 2);40 assert(v1[ 3] == 1);41 assert(v1[ 4] == 1);42 assert(v1[ 5] == 0);43 assert(v1[ 6] == 6);44 assert(v1[ 7] == 0);45 assert(v1[ 8] == 0);46 assert(v1[ 9] == 0);47 assert(v1[10] == 10);48 assert(v1[11] == 0);49 assert(v1[12] == 0);50 assert(v1[13] == 0);51 assert(v1[14] == 14);52 assert(v1[15] == 0);53 assert(v1[16] == 0);54 assert(v1[17] == 0);55 assert(v1[18] == 18);56 assert(v1[19] == 19);57 assert(v1[20] == 20);58 assert(v1[21] == 21);59 assert(v1[22] == 0);60 assert(v1[23] == 0);61 assert(v1[24] == 0);62 assert(v1[25] == 25);63 assert(v1[26] == 0);64 assert(v1[27] == 0);65 assert(v1[28] == 0);66 assert(v1[29] == 29);67 assert(v1[30] == 0);68 assert(v1[31] == 0);69 assert(v1[32] == 0);70 assert(v1[33] == 33);71 assert(v1[34] == 0);72 assert(v1[35] == 0);73 assert(v1[36] == 0);74 assert(v1[37] == 37);75 assert(v1[38] == 38);76 assert(v1[39] == 39);77 assert(v1[40] == 40);78 79 return 0;80}81