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 mask_array |
12 | |
13 | // void operator-=(const valarray<value_type>& v) const; |
14 | |
15 | #include <valarray> |
16 | #include <cassert> |
17 | |
18 | #include "test_macros.h" |
19 | |
20 | int main(int, char**) |
21 | { |
22 | int a1[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}; |
23 | const std::size_t N1 = sizeof(a1)/sizeof(a1[0]); |
24 | int a2[] = {1, 2, 3, 4, 5}; |
25 | const std::size_t N2 = sizeof(a2)/sizeof(a2[0]); |
26 | bool b[N1] = {true, false, false, true, true, false, |
27 | false, true, false, false, false, true}; |
28 | std::valarray<int> v1(a1, N1); |
29 | std::valarray<int> v2(a2, N2); |
30 | assert(N2 == std::count(b, b+N1, true)); |
31 | std::valarray<bool> vb(b, N1); |
32 | v1[vb] -= v2; |
33 | assert(v1.size() == 16); |
34 | assert(v1[ 0] == -1); |
35 | assert(v1[ 1] == 1); |
36 | assert(v1[ 2] == 2); |
37 | assert(v1[ 3] == 1); |
38 | assert(v1[ 4] == 1); |
39 | assert(v1[ 5] == 5); |
40 | assert(v1[ 6] == 6); |
41 | assert(v1[ 7] == 3); |
42 | assert(v1[ 8] == 8); |
43 | assert(v1[ 9] == 9); |
44 | assert(v1[10] == 10); |
45 | assert(v1[11] == 6); |
46 | assert(v1[12] == 12); |
47 | assert(v1[13] == 13); |
48 | assert(v1[14] == 14); |
49 | assert(v1[15] == 15); |
50 | |
51 | return 0; |
52 | } |
53 | |