79 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 valarray;12 13// valarray operator-() const;14 15#include <valarray>16#include <cassert>17#include <cstddef>18 19#include "test_macros.h"20 21int main(int, char**)22{23 {24 typedef int T;25 T a[] = {1, 2, 3, 4, 5};26 const unsigned N = sizeof(a)/sizeof(a[0]);27 std::valarray<T> v(a, N);28 std::valarray<T> v2 = -v;29 assert(v2.size() == v.size());30 for (std::size_t i = 0; i < v2.size(); ++i)31 assert(v2[i] == -v[i]);32 }33 {34 typedef double T;35 T a[] = {1, 2.5, 3, 4.25, 5};36 const unsigned N = sizeof(a)/sizeof(a[0]);37 std::valarray<T> v(a, N);38 std::valarray<T> v2 = -v;39 assert(v2.size() == v.size());40 for (std::size_t i = 0; i < v2.size(); ++i)41 assert(v2[i] == -v[i]);42 }43 {44 typedef std::valarray<double> T;45 T a[] = {T(1), T(2), T(3), T(4), T(5)};46 const unsigned N = sizeof(a)/sizeof(a[0]);47 std::valarray<T> v(a, N);48 std::valarray<T> v2 = -v;49 assert(v2.size() == v.size());50 for (unsigned i = 0; i < N; ++i)51 {52 assert(v2[i].size() == v[i].size());53 for (std::size_t j = 0; j < v[i].size(); ++j)54 assert(v2[i][j] == -v[i][j]);55 }56 }57 {58 typedef double T;59 T a[] = {1, 2.5, 3, 4.25, 5};60 const unsigned N = sizeof(a)/sizeof(a[0]);61 std::valarray<T> v(a, N);62 std::valarray<T> v2 = -(v + v);63 assert(v2.size() == v.size());64 for (std::size_t i = 0; i < v2.size(); ++i)65 assert(v2[i] == -2*v[i]);66 }67 {68 // Make sure we don't have dangling reference problems with unary expressions69 int array[] = {1, 2, 3};70 std::valarray<int> a(array, 3);71 std::valarray<int> b(array, 3);72 auto c = -a * b;73 assert(c.size() == 3);74 assert(c[0] == -1 && c[1] == -4 && c[2] == -9);75 }76 77 return 0;78}79