test/std/numerics/numarray/template.valarray/valarray.cassign/shift_right_valarray.pass.cpp

   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 // <valarray>
  11
  12 // template<class T> class valarray;
  13
  14 // valarray& operator>>=(const valarray& v);
  15
  16 #include <valarray>
  17 #include <cassert>
  18
  19 int main()
  20 {
  21     {
  22         typedef int T;
  23         T a1[] = { 1,   2,   3,    4,    5};
  24         T a2[] = { 6,   7,   8,    9,   10};
  25         T a3[] = {64, 256, 768, 2048, 5120};
  26         const unsigned N = sizeof(a1)/sizeof(a1[0]);
  27         std::valarray<T> v1(a1, N);
  28         std::valarray<T> v2(a2, N);
  29         std::valarray<T> v3(a3, N);
  30         v3 >>= v2;
  31         assert(v1.size() == v2.size());
  32         assert(v1.size() == v3.size());
  33         for (int i = 0; i < v1.size(); ++i)
  34             assert(v1[i] == v3[i]);
  35     }
  36 }