blob: bc721656d2e73f3d14765a57ef023f78dd817b7f [file] [log] [blame]
Gael Guennebaud8cef5412008-06-21 17:28:07 +00001// This file is part of Eigen, a lightweight C++ template library
Benoit Jacob6347b1d2009-05-22 20:25:33 +02002// for linear algebra.
Gael Guennebaud8cef5412008-06-21 17:28:07 +00003//
Gael Guennebaud28e64b02010-06-24 23:21:58 +02004// Copyright (C) 2008-2009 Gael Guennebaud <gael.guennebaud@inria.fr>
Gael Guennebaud8cef5412008-06-21 17:28:07 +00005//
Benoit Jacob69124cf2012-07-13 14:42:47 -04006// This Source Code Form is subject to the terms of the Mozilla
7// Public License v. 2.0. If a copy of the MPL was not distributed
8// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
Gael Guennebaud19bc4182013-02-25 01:17:44 +01009
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000010#include <vector>
Gael Guennebaud8cef5412008-06-21 17:28:07 +000011#include "main.h"
Gael Guennebaud8cef5412008-06-21 17:28:07 +000012
Charles Schlosserb7551bf2022-11-21 20:07:19 +000013template <typename Scalar, std::enable_if_t<NumTraits<Scalar>::IsInteger,int> = 0>
14std::vector<Scalar> special_values() {
15 const Scalar zero = Scalar(0);
16 const Scalar one = Scalar(1);
17 const Scalar two = Scalar(2);
18 const Scalar three = Scalar(3);
19 const Scalar min = (std::numeric_limits<Scalar>::min)();
20 const Scalar max = (std::numeric_limits<Scalar>::max)();
21 return { zero, min, one, two, three, max };
22}
23
24template <typename Scalar, std::enable_if_t<!NumTraits<Scalar>::IsInteger, int> = 0>
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000025std::vector<Scalar> special_values() {
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000026 const Scalar zero = Scalar(0);
Antonio Sanchez8dfe1022021-03-16 20:12:46 -070027 const Scalar eps = Eigen::NumTraits<Scalar>::epsilon();
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000028 const Scalar one = Scalar(1);
Rasmus Munk Larsen6e3b7952021-02-05 16:58:49 -080029 const Scalar two = Scalar(2);
30 const Scalar three = Scalar(3);
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000031 const Scalar sqrt_half = Scalar(std::sqrt(0.5));
32 const Scalar sqrt2 = Scalar(std::sqrt(2));
Antonio Sanchez8dfe1022021-03-16 20:12:46 -070033 const Scalar inf = Eigen::NumTraits<Scalar>::infinity();
34 const Scalar nan = Eigen::NumTraits<Scalar>::quiet_NaN();
Rasmus Munk Larsenbe0574e2021-02-17 02:50:32 +000035 const Scalar denorm_min = std::numeric_limits<Scalar>::denorm_min();
Rasmus Munk Larsen6e3b7952021-02-05 16:58:49 -080036 const Scalar min = (std::numeric_limits<Scalar>::min)();
37 const Scalar max = (std::numeric_limits<Scalar>::max)();
Antonio Sanchez8dfe1022021-03-16 20:12:46 -070038 const Scalar max_exp = (static_cast<Scalar>(int(Eigen::NumTraits<Scalar>::max_exponent())) * Scalar(EIGEN_LN2)) / eps;
Charles Schlosserb7551bf2022-11-21 20:07:19 +000039 return { zero, denorm_min, min, eps, sqrt_half, one, sqrt2, two, three, max_exp, max, inf, nan };
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000040}
41
42template<typename Scalar>
43void special_value_pairs(Array<Scalar, Dynamic, Dynamic>& x,
44 Array<Scalar, Dynamic, Dynamic>& y) {
45 std::vector<Scalar> abs_vals = special_values<Scalar>();
Charles Schlosserb7551bf2022-11-21 20:07:19 +000046 const Index abs_cases = (Index)abs_vals.size();
47 const Index num_cases = 2*abs_cases * 2*abs_cases;
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000048 // ensure both vectorized and non-vectorized paths taken
Charles Schlosserb7551bf2022-11-21 20:07:19 +000049 const Index num_repeats = 2 * (Index)internal::packet_traits<Scalar>::size + 1;
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000050 x.resize(num_repeats, num_cases);
51 y.resize(num_repeats, num_cases);
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000052 int count = 0;
Charles Schlosserb7551bf2022-11-21 20:07:19 +000053 for (Index i = 0; i < abs_cases; ++i) {
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000054 const Scalar abs_x = abs_vals[i];
Charles Schlosserb7551bf2022-11-21 20:07:19 +000055 for (Index sign_x = 0; sign_x < 2; ++sign_x) {
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000056 Scalar x_case = sign_x == 0 ? -abs_x : abs_x;
Charles Schlosserb7551bf2022-11-21 20:07:19 +000057 for (Index j = 0; j < abs_cases; ++j) {
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000058 const Scalar abs_y = abs_vals[j];
Charles Schlosserb7551bf2022-11-21 20:07:19 +000059 for (Index sign_y = 0; sign_y < 2; ++sign_y) {
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000060 Scalar y_case = sign_y == 0 ? -abs_y : abs_y;
Charles Schlosserb7551bf2022-11-21 20:07:19 +000061 for (Index repeat = 0; repeat < num_repeats; ++repeat) {
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000062 x(repeat, count) = x_case;
63 y(repeat, count) = y_case;
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000064 }
65 ++count;
66 }
67 }
68 }
69 }
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000070}
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000071
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000072template <typename Scalar, typename Fn, typename RefFn>
73void binary_op_test(std::string name, Fn fun, RefFn ref) {
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000074 const Scalar tol = test_precision<Scalar>();
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000075 Array<Scalar, Dynamic, Dynamic> x;
76 Array<Scalar, Dynamic, Dynamic> y;
77 special_value_pairs(x, y);
78
79 Array<Scalar, Dynamic, Dynamic> actual = fun(x, y);
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000080 bool all_pass = true;
Charles Schlosserb7551bf2022-11-21 20:07:19 +000081 for (Index i = 0; i < x.rows(); ++i) {
82 for (Index j = 0; j < x.cols(); ++j) {
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000083 Scalar e = static_cast<Scalar>(ref(x(i,j), y(i,j)));
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000084 Scalar a = actual(i, j);
Rasmus Munk Larsen7a87ed12022-08-08 18:48:36 +000085 bool success = (a==e) || ((numext::isfinite)(e) && internal::isApprox(a, e, tol)) || ((numext::isnan)(a) && (numext::isnan)(e));
86 all_pass &= success;
87 if (!success) {
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000088 std::cout << name << "(" << x(i,j) << "," << y(i,j) << ") = " << a << " != " << e << std::endl;
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +000089 }
90 }
91 }
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000092 VERIFY(all_pass);
93}
Charles Schlosser76a669f2022-08-16 21:32:36 +000094
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +000095template <typename Scalar>
96void binary_ops_test() {
97 binary_op_test<Scalar>("pow",
98 [](auto x, auto y) { return Eigen::pow(x, y); },
99 [](auto x, auto y) { return std::pow(x, y); });
100 binary_op_test<Scalar>("atan2",
101 [](auto x, auto y) { return Eigen::atan2(x, y); },
102 [](auto x, auto y) { return std::atan2(x, y); });
103}
Charles Schlosser76a669f2022-08-16 21:32:36 +0000104
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +0000105template <typename Scalar>
106void pow_scalar_exponent_test() {
107 using Int_t = typename internal::make_integer<Scalar>::type;
108 const Scalar tol = test_precision<Scalar>();
109
110 std::vector<Scalar> abs_vals = special_values<Scalar>();
Charles Schlosserb7551bf2022-11-21 20:07:19 +0000111 const Index num_vals = (Index)abs_vals.size();
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +0000112 Map<Array<Scalar, Dynamic, 1>> bases(abs_vals.data(), num_vals);
113
114 bool all_pass = true;
115 for (Scalar abs_exponent : abs_vals) {
116 for (Scalar exponent : {-abs_exponent, abs_exponent}) {
117 // test integer exponent code path
118 bool exponent_is_integer = (numext::isfinite)(exponent) && (numext::round(exponent) == exponent) &&
119 (numext::abs(exponent) < static_cast<Scalar>(NumTraits<Int_t>::highest()));
120 if (exponent_is_integer) {
121 Int_t exponent_as_int = static_cast<Int_t>(exponent);
122 Array<Scalar, Dynamic, 1> eigenPow = bases.pow(exponent_as_int);
Charles Schlosserb7551bf2022-11-21 20:07:19 +0000123 for (Index j = 0; j < num_vals; j++) {
Charles Schlosser76a669f2022-08-16 21:32:36 +0000124 Scalar e = static_cast<Scalar>(std::pow(bases(j), exponent));
125 Scalar a = eigenPow(j);
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +0000126 bool success = (a == e) || ((numext::isfinite)(e) && internal::isApprox(a, e, tol)) ||
127 ((numext::isnan)(a) && (numext::isnan)(e));
Charles Schlosser76a669f2022-08-16 21:32:36 +0000128 all_pass &= success;
129 if (!success) {
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +0000130 std::cout << "pow(" << bases(j) << "," << exponent << ") = " << a << " != " << e << std::endl;
Charles Schlosser76a669f2022-08-16 21:32:36 +0000131 }
132 }
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +0000133 } else {
134 // test floating point exponent code path
135 Array<Scalar, Dynamic, 1> eigenPow = bases.pow(exponent);
Charles Schlosserb7551bf2022-11-21 20:07:19 +0000136 for (Index j = 0; j < num_vals; j++) {
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +0000137 Scalar e = static_cast<Scalar>(std::pow(bases(j), exponent));
138 Scalar a = eigenPow(j);
139 bool success = (a == e) || ((numext::isfinite)(e) && internal::isApprox(a, e, tol)) ||
140 ((numext::isnan)(a) && (numext::isnan)(e));
141 all_pass &= success;
142 if (!success) {
143 std::cout << "pow(" << bases(j) << "," << exponent << ") = " << a << " != " << e << std::endl;
Charles Schlosser76a669f2022-08-16 21:32:36 +0000144 }
145 }
146 }
147 }
148 }
Rasmus Munk Larsencdd8fdc2021-01-18 13:25:16 +0000149 VERIFY(all_pass);
150}
151
Charles Schlossere5af9f82022-08-29 19:23:54 +0000152template <typename Scalar, typename ScalarExponent>
153Scalar calc_overflow_threshold(const ScalarExponent exponent) {
154 EIGEN_USING_STD(exp2);
Antonio Sánchez69f50e32022-09-06 19:53:29 +0000155 EIGEN_USING_STD(log2);
Charles Schlossere5af9f82022-08-29 19:23:54 +0000156 EIGEN_STATIC_ASSERT((NumTraits<Scalar>::digits() < 2 * NumTraits<double>::digits()), BASE_TYPE_IS_TOO_BIG);
157
158 if (exponent < 2)
159 return NumTraits<Scalar>::highest();
160 else {
Antonio Sánchez69f50e32022-09-06 19:53:29 +0000161 // base^e <= highest ==> base <= 2^(log2(highest)/e)
Antonio Sánchez3c37dd22022-09-08 17:40:45 +0000162 // For floating-point types, consider the bound for integer values that can be reproduced exactly = 2 ^ digits
163 double highest_bits = numext::mini(static_cast<double>(NumTraits<Scalar>::digits()),
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +0000164 static_cast<double>(log2(NumTraits<Scalar>::highest())));
Antonio Sánchez69f50e32022-09-06 19:53:29 +0000165 return static_cast<Scalar>(
Antonio Sánchez3c37dd22022-09-08 17:40:45 +0000166 numext::floor(exp2(highest_bits / static_cast<double>(exponent))));
Charles Schlossere5af9f82022-08-29 19:23:54 +0000167 }
168}
169
Rasmus Munk Larsendceb7792022-09-12 15:51:27 -0700170template <typename Base, typename Exponent, bool ExpIsInteger = NumTraits<Exponent>::IsInteger>
171struct ref_pow {
172 static Base run(Base base, Exponent exponent) {
173 EIGEN_USING_STD(pow);
174 return pow(base, static_cast<Base>(exponent));
175 }
176};
177
178template <typename Base, typename Exponent>
179struct ref_pow<Base, Exponent, true> {
180 static Base run(Base base, Exponent exponent) {
181 EIGEN_USING_STD(pow);
182 return pow(base, exponent);
183 }
184};
185
Charles Schlossere5af9f82022-08-29 19:23:54 +0000186template <typename Base, typename Exponent>
187void test_exponent(Exponent exponent) {
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +0000188 const Base max_abs_bases = static_cast<Base>(10000);
189 // avoid integer overflow in Base type
190 Base threshold = calc_overflow_threshold<Base, Exponent>(numext::abs(exponent));
191 // avoid numbers that can't be verified with std::pow
192 double double_threshold = calc_overflow_threshold<double, Exponent>(numext::abs(exponent));
193 // use the lesser of these two thresholds
194 Base testing_threshold =
195 static_cast<double>(threshold) < double_threshold ? threshold : static_cast<Base>(double_threshold);
196 // test both vectorized and non-vectorized code paths
197 const Index array_size = 2 * internal::packet_traits<Base>::size + 1;
198
199 Base max_base = numext::mini(testing_threshold, max_abs_bases);
200 Base min_base = NumTraits<Base>::IsSigned ? -max_base : Base(0);
201
202 ArrayX<Base> x(array_size), y(array_size);
203 bool all_pass = true;
204 for (Base base = min_base; base <= max_base; base++) {
205 if (exponent < 0 && base == 0) continue;
206 x.setConstant(base);
207 y = x.pow(exponent);
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +0000208 for (Base a : y) {
Rasmus Munk Larsendceb7792022-09-12 15:51:27 -0700209 Base e = ref_pow<Base, Exponent>::run(base, exponent);
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +0000210 bool pass = (a == e);
211 if (!NumTraits<Base>::IsInteger) {
212 pass = pass || (((numext::isfinite)(e) && internal::isApprox(a, e)) ||
213 ((numext::isnan)(a) && (numext::isnan)(e)));
214 }
215 all_pass &= pass;
216 if (!pass) {
217 std::cout << "pow(" << base << "," << exponent << ") = " << a << " != " << e << std::endl;
218 }
Charles Schlossere5af9f82022-08-29 19:23:54 +0000219 }
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +0000220 }
221 VERIFY(all_pass);
Charles Schlossere5af9f82022-08-29 19:23:54 +0000222}
Charles Schlossere5af9f82022-08-29 19:23:54 +0000223
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +0000224template <typename Base, typename Exponent>
225void unary_pow_test() {
226 Exponent max_exponent = static_cast<Exponent>(NumTraits<Base>::digits());
227 Exponent min_exponent = static_cast<Exponent>(NumTraits<Exponent>::IsSigned ? -max_exponent : 0);
228
229 for (Exponent exponent = min_exponent; exponent < max_exponent; ++exponent) {
230 test_exponent<Base, Exponent>(exponent);
231 }
Charles Schlosser82b152d2022-11-04 00:31:20 +0000232}
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +0000233
234void mixed_pow_test() {
235 // The following cases will test promoting a smaller exponent type
236 // to a wider base type.
237 unary_pow_test<double, int>();
238 unary_pow_test<double, float>();
239 unary_pow_test<float, half>();
240 unary_pow_test<double, half>();
241 unary_pow_test<float, bfloat16>();
242 unary_pow_test<double, bfloat16>();
243
244 // Although in the following cases the exponent cannot be represented exactly
245 // in the base type, we do not perform a conversion, but implement
246 // the operation using repeated squaring.
247 unary_pow_test<float, int>();
248 unary_pow_test<double, long long>();
249
250 // The following cases will test promoting a wider exponent type
251 // to a narrower base type. This should compile but generate a
252 // deprecation warning:
253 unary_pow_test<float, double>();
254}
255
256void int_pow_test() {
257 unary_pow_test<int, int>();
258 unary_pow_test<unsigned int, unsigned int>();
259 unary_pow_test<long long, long long>();
260 unary_pow_test<unsigned long long, unsigned long long>();
261
262 // Although in the following cases the exponent cannot be represented exactly
263 // in the base type, we do not perform a conversion, but implement the
264 // operation using repeated squaring.
265 unary_pow_test<long long, int>();
266 unary_pow_test<int, unsigned int>();
267 unary_pow_test<unsigned int, int>();
268 unary_pow_test<long long, unsigned long long>();
269 unary_pow_test<unsigned long long, long long>();
270 unary_pow_test<long long, int>();
Charles Schlossere5af9f82022-08-29 19:23:54 +0000271}
272
Charles Schlosser82b152d2022-11-04 00:31:20 +0000273namespace Eigen {
274namespace internal {
275template <typename Scalar>
276struct test_signbit_op {
277 Scalar constexpr operator()(const Scalar& a) const { return numext::signbit(a); }
278 template <typename Packet>
279 inline Packet packetOp(const Packet& a) const {
280 return psignbit(a);
281 }
282};
283template <typename Scalar>
284struct functor_traits<test_signbit_op<Scalar>> {
285 enum { Cost = 1, PacketAccess = true }; //todo: define HasSignbit flag
286};
287} // namespace internal
288} // namespace Eigen
289
290template <typename T, bool IsInteger = NumTraits<T>::IsInteger>
291struct ref_signbit_func_impl {
292 static bool run(const T& x) { return std::signbit(x); }
293};
294template <typename T>
295struct ref_signbit_func_impl<T, true> {
296 // MSVC (perhaps others) does not have a std::signbit overload for integers
297 static bool run(const T& x) { return x < T(0); }
298};
299template <typename T>
300bool ref_signbit_func(const T& x) {
301 return ref_signbit_func_impl<T>::run(x);
302}
303
304template <typename Scalar>
305void signbit_test() {
306 Scalar true_mask;
307 std::memset(static_cast<void*>(&true_mask), 0xff, sizeof(Scalar));
308 Scalar false_mask;
309 std::memset(static_cast<void*>(&false_mask), 0x00, sizeof(Scalar));
310
311 const size_t size = 100 * internal::packet_traits<Scalar>::size;
312 ArrayX<Scalar> x(size), y(size);
313 x.setRandom();
314 std::vector<Scalar> special_vals = special_values<Scalar>();
315 for (size_t i = 0; i < special_vals.size(); i++) {
316 x(2 * i + 0) = special_vals[i];
317 x(2 * i + 1) = -special_vals[i];
318 }
319 y = x.unaryExpr(internal::test_signbit_op<Scalar>());
320
321 bool all_pass = true;
322 for (size_t i = 0; i < size; i++) {
323 const Scalar ref_val = ref_signbit_func(x(i)) ? true_mask : false_mask;
324 bool not_same = internal::predux_any(internal::bitwise_helper<Scalar>::bitwise_xor(ref_val, y(i)));
325 if (not_same) std::cout << "signbit(" << x(i) << ") != " << y(i) << "\n";
326 all_pass = all_pass && !not_same;
327 }
328
329 VERIFY(all_pass);
330}
331void signbit_tests() {
332 signbit_test<float>();
333 signbit_test<double>();
334 signbit_test<Eigen::half>();
335 signbit_test<Eigen::bfloat16>();
336
337 signbit_test<uint8_t>();
338 signbit_test<uint16_t>();
339 signbit_test<uint32_t>();
340 signbit_test<uint64_t>();
341
342 signbit_test<int8_t>();
343 signbit_test<int16_t>();
344 signbit_test<int32_t>();
345 signbit_test<int64_t>();
346}
347
Benoit Jacobe2775862010-04-28 18:51:38 -0400348template<typename ArrayType> void array(const ArrayType& m)
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000349{
Benoit Jacobe2775862010-04-28 18:51:38 -0400350 typedef typename ArrayType::Scalar Scalar;
Gael Guennebaudd476cad2016-06-25 10:12:06 +0200351 typedef typename ArrayType::RealScalar RealScalar;
Benoit Jacobe2775862010-04-28 18:51:38 -0400352 typedef Array<Scalar, ArrayType::RowsAtCompileTime, 1> ColVectorType;
353 typedef Array<Scalar, 1, ArrayType::ColsAtCompileTime> RowVectorType;
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000354
Hauke Heibelf1679c72010-06-20 17:37:56 +0200355 Index rows = m.rows();
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800356 Index cols = m.cols();
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000357
Rasmus Munk Larsen98e51c92022-08-26 16:19:03 +0000358 ArrayType m1 = ArrayType::Random(rows, cols);
359 if (NumTraits<RealScalar>::IsInteger && NumTraits<RealScalar>::IsSigned
360 && !NumTraits<Scalar>::IsComplex) {
361 // Here we cap the size of the values in m1 such that pow(3)/cube()
362 // doesn't overflow and result in undefined behavior. Notice that because
363 // pow(int, int) promotes its inputs and output to double (according to
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +0000364 // the C++ standard), we have to make sure that the result fits in 53 bits
Rasmus Munk Larsen98e51c92022-08-26 16:19:03 +0000365 // for int64,
366 RealScalar max_val =
367 numext::mini(RealScalar(std::cbrt(NumTraits<RealScalar>::highest())),
368 RealScalar(std::cbrt(1LL << 53)))/2;
369 m1.array() = (m1.abs().array() <= max_val).select(m1, Scalar(max_val));
370 }
371 ArrayType m2 = ArrayType::Random(rows, cols),
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000372 m3(rows, cols);
Christoph Hertzberg3be9f5c2015-04-16 13:25:20 +0200373 ArrayType m4 = m1; // copy constructor
374 VERIFY_IS_APPROX(m1, m4);
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000375
Gael Guennebaud627595a2009-06-10 11:20:30 +0200376 ColVectorType cv1 = ColVectorType::Random(rows);
377 RowVectorType rv1 = RowVectorType::Random(cols);
378
Benoit Jacob47160402010-10-25 10:15:22 -0400379 Scalar s1 = internal::random<Scalar>(),
Hauke Heibel8cb3e362012-03-02 16:27:27 +0100380 s2 = internal::random<Scalar>();
381
Gael Guennebaud59dc1da2008-09-03 17:16:28 +0000382 // scalar addition
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100383 VERIFY_IS_APPROX(m1 + s1, s1 + m1);
Benoit Jacobe2775862010-04-28 18:51:38 -0400384 VERIFY_IS_APPROX(m1 + s1, ArrayType::Constant(rows,cols,s1) + m1);
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100385 VERIFY_IS_APPROX(s1 - m1, (-m1)+s1 );
Benoit Jacobe2775862010-04-28 18:51:38 -0400386 VERIFY_IS_APPROX(m1 - s1, m1 - ArrayType::Constant(rows,cols,s1));
387 VERIFY_IS_APPROX(s1 - m1, ArrayType::Constant(rows,cols,s1) - m1);
388 VERIFY_IS_APPROX((m1*Scalar(2)) - s2, (m1+m1) - ArrayType::Constant(rows,cols,s2) );
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000389 m3 = m1;
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100390 m3 += s2;
391 VERIFY_IS_APPROX(m3, m1 + s2);
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000392 m3 = m1;
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100393 m3 -= s1;
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800394 VERIFY_IS_APPROX(m3, m1 - s1);
395
Hauke Heibelf578dc72010-12-16 17:34:13 +0100396 // scalar operators via Maps
Rasmus Munk Larsen98e51c92022-08-26 16:19:03 +0000397 m3 = m1; m4 = m1;
398 ArrayType::Map(m4.data(), m4.rows(), m4.cols()) -= ArrayType::Map(m2.data(), m2.rows(), m2.cols());
399 VERIFY_IS_APPROX(m4, m3 - m2);
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800400
Rasmus Munk Larsen98e51c92022-08-26 16:19:03 +0000401 m3 = m1; m4 = m1;
402 ArrayType::Map(m4.data(), m4.rows(), m4.cols()) += ArrayType::Map(m2.data(), m2.rows(), m2.cols());
403 VERIFY_IS_APPROX(m4, m3 + m2);
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800404
Rasmus Munk Larsen98e51c92022-08-26 16:19:03 +0000405 m3 = m1; m4 = m1;
406 ArrayType::Map(m4.data(), m4.rows(), m4.cols()) *= ArrayType::Map(m2.data(), m2.rows(), m2.cols());
407 VERIFY_IS_APPROX(m4, m3 * m2);
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800408
Rasmus Munk Larsen98e51c92022-08-26 16:19:03 +0000409 m3 = m1; m4 = m1;
Hauke Heibelf578dc72010-12-16 17:34:13 +0100410 m2 = ArrayType::Random(rows,cols);
411 m2 = (m2==0).select(1,m2);
Rasmus Munk Larsen98e51c92022-08-26 16:19:03 +0000412 ArrayType::Map(m4.data(), m4.rows(), m4.cols()) /= ArrayType::Map(m2.data(), m2.rows(), m2.cols());
413 VERIFY_IS_APPROX(m4, m3 / m2);
Gael Guennebaud05ad0832008-07-19 13:03:23 +0000414
Gael Guennebaud59dc1da2008-09-03 17:16:28 +0000415 // reductions
Gael Guennebaud42af5872013-02-23 22:58:14 +0100416 VERIFY_IS_APPROX(m1.abs().colwise().sum().sum(), m1.abs().sum());
417 VERIFY_IS_APPROX(m1.abs().rowwise().sum().sum(), m1.abs().sum());
418 using std::abs;
419 VERIFY_IS_MUCH_SMALLER_THAN(abs(m1.colwise().sum().sum() - m1.sum()), m1.abs().sum());
420 VERIFY_IS_MUCH_SMALLER_THAN(abs(m1.rowwise().sum().sum() - m1.sum()), m1.abs().sum());
Gael Guennebaud28e139a2013-02-23 23:06:45 +0100421 if (!internal::isMuchSmallerThan(abs(m1.sum() - (m1+m2).sum()), m1.abs().sum(), test_precision<Scalar>()))
Hauke Heibel83e3c452010-12-16 18:53:02 +0100422 VERIFY_IS_NOT_APPROX(((m1+m2).rowwise().sum()).sum(), m1.sum());
Gael Guennebaud66e99ab2016-06-06 15:11:41 +0200423 VERIFY_IS_APPROX(m1.colwise().sum(), m1.colwise().redux(internal::scalar_sum_op<Scalar,Scalar>()));
Gael Guennebaud627595a2009-06-10 11:20:30 +0200424
425 // vector-wise ops
426 m3 = m1;
427 VERIFY_IS_APPROX(m3.colwise() += cv1, m1.colwise() + cv1);
428 m3 = m1;
429 VERIFY_IS_APPROX(m3.colwise() -= cv1, m1.colwise() - cv1);
430 m3 = m1;
431 VERIFY_IS_APPROX(m3.rowwise() += rv1, m1.rowwise() + rv1);
432 m3 = m1;
433 VERIFY_IS_APPROX(m3.rowwise() -= rv1, m1.rowwise() - rv1);
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800434
Gael Guennebaud0a18eec2014-09-19 13:25:28 +0200435 // Conversion from scalar
436 VERIFY_IS_APPROX((m3 = s1), ArrayType::Constant(rows,cols,s1));
437 VERIFY_IS_APPROX((m3 = 1), ArrayType::Constant(rows,cols,1));
438 VERIFY_IS_APPROX((m3.topLeftCorner(rows,cols) = 1), ArrayType::Constant(rows,cols,1));
439 typedef Array<Scalar,
440 ArrayType::RowsAtCompileTime==Dynamic?2:ArrayType::RowsAtCompileTime,
441 ArrayType::ColsAtCompileTime==Dynamic?2:ArrayType::ColsAtCompileTime,
442 ArrayType::Options> FixedArrayType;
Gael Guennebaud543529d2019-01-22 15:30:50 +0100443 {
444 FixedArrayType f1(s1);
445 VERIFY_IS_APPROX(f1, FixedArrayType::Constant(s1));
446 FixedArrayType f2(numext::real(s1));
447 VERIFY_IS_APPROX(f2, FixedArrayType::Constant(numext::real(s1)));
448 FixedArrayType f3((int)100*numext::real(s1));
449 VERIFY_IS_APPROX(f3, FixedArrayType::Constant((int)100*numext::real(s1)));
450 f1.setRandom();
451 FixedArrayType f4(f1.data());
452 VERIFY_IS_APPROX(f4, f1);
453 }
Gael Guennebaud543529d2019-01-22 15:30:50 +0100454 {
455 FixedArrayType f1{s1};
456 VERIFY_IS_APPROX(f1, FixedArrayType::Constant(s1));
457 FixedArrayType f2{numext::real(s1)};
458 VERIFY_IS_APPROX(f2, FixedArrayType::Constant(numext::real(s1)));
459 FixedArrayType f3{(int)100*numext::real(s1)};
460 VERIFY_IS_APPROX(f3, FixedArrayType::Constant((int)100*numext::real(s1)));
461 f1.setRandom();
462 FixedArrayType f4{f1.data()};
463 VERIFY_IS_APPROX(f4, f1);
464 }
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800465
Gael Guennebaudd476cad2016-06-25 10:12:06 +0200466 // pow
467 VERIFY_IS_APPROX(m1.pow(2), m1.square());
468 VERIFY_IS_APPROX(pow(m1,2), m1.square());
469 VERIFY_IS_APPROX(m1.pow(3), m1.cube());
470 VERIFY_IS_APPROX(pow(m1,3), m1.cube());
471 VERIFY_IS_APPROX((-m1).pow(3), -m1.cube());
472 VERIFY_IS_APPROX(pow(2*m1,3), 8*m1.cube());
473 ArrayType exponents = ArrayType::Constant(rows, cols, RealScalar(2));
474 VERIFY_IS_APPROX(Eigen::pow(m1,exponents), m1.square());
475 VERIFY_IS_APPROX(m1.pow(exponents), m1.square());
476 VERIFY_IS_APPROX(Eigen::pow(2*m1,exponents), 4*m1.square());
477 VERIFY_IS_APPROX((2*m1).pow(exponents), 4*m1.square());
478 VERIFY_IS_APPROX(Eigen::pow(m1,2*exponents), m1.square().square());
479 VERIFY_IS_APPROX(m1.pow(2*exponents), m1.square().square());
Gael Guennebaude61cee72016-07-04 11:49:03 +0200480 VERIFY_IS_APPROX(Eigen::pow(m1(0,0), exponents), ArrayType::Constant(rows,cols,m1(0,0)*m1(0,0)));
Gael Guennebaudd476cad2016-06-25 10:12:06 +0200481
Gael Guennebaud0a18eec2014-09-19 13:25:28 +0200482 // Check possible conflicts with 1D ctor
483 typedef Array<Scalar, Dynamic, 1> OneDArrayType;
Gael Guennebaud543529d2019-01-22 15:30:50 +0100484 {
485 OneDArrayType o1(rows);
486 VERIFY(o1.size()==rows);
487 OneDArrayType o2(static_cast<int>(rows));
488 VERIFY(o2.size()==rows);
489 }
Gael Guennebaud543529d2019-01-22 15:30:50 +0100490 {
491 OneDArrayType o1{rows};
492 VERIFY(o1.size()==rows);
493 OneDArrayType o4{int(rows)};
494 VERIFY(o4.size()==rows);
495 }
Gael Guennebaud543529d2019-01-22 15:30:50 +0100496 // Check possible conflicts with 2D ctor
497 typedef Array<Scalar, Dynamic, Dynamic> TwoDArrayType;
498 typedef Array<Scalar, 2, 1> ArrayType2;
499 {
500 TwoDArrayType o1(rows,cols);
501 VERIFY(o1.rows()==rows);
502 VERIFY(o1.cols()==cols);
503 TwoDArrayType o2(static_cast<int>(rows),static_cast<int>(cols));
504 VERIFY(o2.rows()==rows);
505 VERIFY(o2.cols()==cols);
506
507 ArrayType2 o3(rows,cols);
508 VERIFY(o3(0)==Scalar(rows) && o3(1)==Scalar(cols));
509 ArrayType2 o4(static_cast<int>(rows),static_cast<int>(cols));
510 VERIFY(o4(0)==Scalar(rows) && o4(1)==Scalar(cols));
511 }
Gael Guennebaud543529d2019-01-22 15:30:50 +0100512 {
513 TwoDArrayType o1{rows,cols};
514 VERIFY(o1.rows()==rows);
515 VERIFY(o1.cols()==cols);
516 TwoDArrayType o2{int(rows),int(cols)};
517 VERIFY(o2.rows()==rows);
518 VERIFY(o2.cols()==cols);
519
520 ArrayType2 o3{rows,cols};
521 VERIFY(o3(0)==Scalar(rows) && o3(1)==Scalar(cols));
522 ArrayType2 o4{int(rows),int(cols)};
523 VERIFY(o4(0)==Scalar(rows) && o4(1)==Scalar(cols));
524 }
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000525}
526
Benoit Jacobe2775862010-04-28 18:51:38 -0400527template<typename ArrayType> void comparisons(const ArrayType& m)
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000528{
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100529 using std::abs;
Benoit Jacobe2775862010-04-28 18:51:38 -0400530 typedef typename ArrayType::Scalar Scalar;
Benoit Jacob874ff5a2009-01-26 17:56:04 +0000531 typedef typename NumTraits<Scalar>::Real RealScalar;
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000532
Hauke Heibelf1679c72010-06-20 17:37:56 +0200533 Index rows = m.rows();
534 Index cols = m.cols();
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000535
Benoit Jacob47160402010-10-25 10:15:22 -0400536 Index r = internal::random<Index>(0, rows-1),
537 c = internal::random<Index>(0, cols-1);
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000538
Benoit Jacobe2775862010-04-28 18:51:38 -0400539 ArrayType m1 = ArrayType::Random(rows, cols),
Gael Guennebaud9fc1c922015-06-22 16:48:27 +0200540 m2 = ArrayType::Random(rows, cols),
541 m3(rows, cols),
542 m4 = m1;
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800543
Gael Guennebaud9fc1c922015-06-22 16:48:27 +0200544 m4 = (m4.abs()==Scalar(0)).select(1,m4);
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000545
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100546 VERIFY(((m1 + Scalar(1)) > m1).all());
547 VERIFY(((m1 - Scalar(1)) < m1).all());
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000548 if (rows*cols>1)
549 {
550 m3 = m1;
551 m3(r,c) += 1;
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100552 VERIFY(! (m1 < m3).all() );
553 VERIFY(! (m1 > m3).all() );
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000554 }
Christoph Hertzberg494fa992015-05-07 17:28:40 +0200555 VERIFY(!(m1 > m2 && m1 < m2).any());
556 VERIFY((m1 <= m2 || m1 >= m2).all());
Gael Guennebaud627595a2009-06-10 11:20:30 +0200557
Christoph Hertzberg36052c42013-10-17 15:37:29 +0200558 // comparisons array to scalar
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100559 VERIFY( (m1 != (m1(r,c)+1) ).any() );
Christoph Hertzberg36052c42013-10-17 15:37:29 +0200560 VERIFY( (m1 > (m1(r,c)-1) ).any() );
561 VERIFY( (m1 < (m1(r,c)+1) ).any() );
562 VERIFY( (m1 == m1(r,c) ).any() );
563
564 // comparisons scalar to array
565 VERIFY( ( (m1(r,c)+1) != m1).any() );
566 VERIFY( ( (m1(r,c)-1) < m1).any() );
567 VERIFY( ( (m1(r,c)+1) > m1).any() );
568 VERIFY( ( m1(r,c) == m1).any() );
Gael Guennebaud627595a2009-06-10 11:20:30 +0200569
Gael Guennebaud59dc1da2008-09-03 17:16:28 +0000570 // test Select
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100571 VERIFY_IS_APPROX( (m1<m2).select(m1,m2), m1.cwiseMin(m2) );
572 VERIFY_IS_APPROX( (m1>m2).select(m1,m2), m1.cwiseMax(m2) );
Gael Guennebaud9f795582009-12-16 19:18:40 +0100573 Scalar mid = (m1.cwiseAbs().minCoeff() + m1.cwiseAbs().maxCoeff())/Scalar(2);
Gael Guennebaud59dc1da2008-09-03 17:16:28 +0000574 for (int j=0; j<cols; ++j)
575 for (int i=0; i<rows; ++i)
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100576 m3(i,j) = abs(m1(i,j))<mid ? 0 : m1(i,j);
Benoit Jacobe2775862010-04-28 18:51:38 -0400577 VERIFY_IS_APPROX( (m1.abs()<ArrayType::Constant(rows,cols,mid))
578 .select(ArrayType::Zero(rows,cols),m1), m3);
Gael Guennebaude14aa8c2008-09-03 17:56:06 +0000579 // shorter versions:
Benoit Jacobe2775862010-04-28 18:51:38 -0400580 VERIFY_IS_APPROX( (m1.abs()<ArrayType::Constant(rows,cols,mid))
Gael Guennebaud59dc1da2008-09-03 17:16:28 +0000581 .select(0,m1), m3);
Benoit Jacobe2775862010-04-28 18:51:38 -0400582 VERIFY_IS_APPROX( (m1.abs()>=ArrayType::Constant(rows,cols,mid))
Gael Guennebaud59dc1da2008-09-03 17:16:28 +0000583 .select(m1,0), m3);
Gael Guennebaude14aa8c2008-09-03 17:56:06 +0000584 // even shorter version:
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100585 VERIFY_IS_APPROX( (m1.abs()<mid).select(0,m1), m3);
Gael Guennebaud627595a2009-06-10 11:20:30 +0200586
Gael Guennebaud56c7e162009-01-24 15:22:44 +0000587 // count
Gael Guennebaudc70d5422010-01-18 22:54:20 +0100588 VERIFY(((m1.abs()+1)>RealScalar(0.1)).count() == rows*cols);
Benoit Jacobaaaade42010-05-30 16:00:58 -0400589
Gael Guennebaud35c11582011-05-31 22:17:34 +0200590 // and/or
591 VERIFY( (m1<RealScalar(0) && m1>RealScalar(0)).count() == 0);
592 VERIFY( (m1<RealScalar(0) || m1>=RealScalar(0)).count() == rows*cols);
593 RealScalar a = m1.abs().mean();
594 VERIFY( (m1<-a || m1>a).count() == (m1.abs()>a).count());
595
Gael Guennebaud12e1ebb2018-07-12 17:16:40 +0200596 typedef Array<Index, Dynamic, 1> ArrayOfIndices;
Gael Guennebaud4ebb8042010-06-02 09:45:57 +0200597
Gael Guennebaud9f795582009-12-16 19:18:40 +0100598 // TODO allows colwise/rowwise for array
Benoit Jacobaaaade42010-05-30 16:00:58 -0400599 VERIFY_IS_APPROX(((m1.abs()+1)>RealScalar(0.1)).colwise().count(), ArrayOfIndices::Constant(cols,rows).transpose());
600 VERIFY_IS_APPROX(((m1.abs()+1)>RealScalar(0.1)).rowwise().count(), ArrayOfIndices::Constant(rows, cols));
Benoit Jacobe8009992008-11-03 22:47:00 +0000601}
602
Benoit Jacobe2775862010-04-28 18:51:38 -0400603template<typename ArrayType> void array_real(const ArrayType& m)
Gael Guennebaud0ce5bc02010-01-27 23:23:59 +0100604{
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100605 using std::abs;
Gael Guennebaud9b9177f2013-07-05 13:35:34 +0200606 using std::sqrt;
Benoit Jacobe2775862010-04-28 18:51:38 -0400607 typedef typename ArrayType::Scalar Scalar;
Gael Guennebaud0ce5bc02010-01-27 23:23:59 +0100608 typedef typename NumTraits<Scalar>::Real RealScalar;
609
Hauke Heibelf1679c72010-06-20 17:37:56 +0200610 Index rows = m.rows();
611 Index cols = m.cols();
Gael Guennebaud0ce5bc02010-01-27 23:23:59 +0100612
Benoit Jacobe2775862010-04-28 18:51:38 -0400613 ArrayType m1 = ArrayType::Random(rows, cols),
Gael Guennebaudc695cbf2013-06-24 13:33:44 +0200614 m2 = ArrayType::Random(rows, cols),
Gael Guennebaud9fc1c922015-06-22 16:48:27 +0200615 m3(rows, cols),
616 m4 = m1;
Benoit Steiner83149622015-12-10 13:13:45 -0800617
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800618 m4 = (m4.abs()==Scalar(0)).select(Scalar(1),m4);
Gael Guennebaud0ce5bc02010-01-27 23:23:59 +0100619
Gael Guennebaud9b1ad5e2012-03-09 12:08:06 +0100620 Scalar s1 = internal::random<Scalar>();
621
Deanna Hood41b717d2015-03-18 03:11:03 +1000622 // these tests are mostly to check possible compilation issues with free-functions.
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100623 VERIFY_IS_APPROX(m1.sin(), sin(m1));
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100624 VERIFY_IS_APPROX(m1.cos(), cos(m1));
Deanna Hoodf89fcef2015-03-11 13:13:30 +1000625 VERIFY_IS_APPROX(m1.tan(), tan(m1));
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100626 VERIFY_IS_APPROX(m1.asin(), asin(m1));
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100627 VERIFY_IS_APPROX(m1.acos(), acos(m1));
Jitse Niesende150b12014-06-19 15:12:33 +0100628 VERIFY_IS_APPROX(m1.atan(), atan(m1));
Deanna Hoodf89fcef2015-03-11 13:13:30 +1000629 VERIFY_IS_APPROX(m1.sinh(), sinh(m1));
630 VERIFY_IS_APPROX(m1.cosh(), cosh(m1));
631 VERIFY_IS_APPROX(m1.tanh(), tanh(m1));
Rasmus Munk Larsen1e1848f2022-09-28 20:46:49 +0000632 VERIFY_IS_APPROX(m1.atan2(m2), atan2(m1,m2));
633
Rasmus Munk Larsen28ba1b22019-01-11 17:45:37 -0800634 VERIFY_IS_APPROX(m1.tanh().atanh(), atanh(tanh(m1)));
635 VERIFY_IS_APPROX(m1.sinh().asinh(), asinh(sinh(m1)));
636 VERIFY_IS_APPROX(m1.cosh().acosh(), acosh(cosh(m1)));
Rasmus Munk Larsend6e283b2018-08-13 11:14:50 -0700637 VERIFY_IS_APPROX(m1.logistic(), logistic(m1));
Gael Guennebaud2f7e2612016-07-08 11:13:55 +0200638
Deanna Hooda5e49972015-03-11 08:56:42 +1000639 VERIFY_IS_APPROX(m1.arg(), arg(m1));
Deanna Hood31fdd672015-03-11 06:39:23 +1000640 VERIFY_IS_APPROX(m1.round(), round(m1));
Ilya Tokar19876ce2019-12-16 16:00:35 -0500641 VERIFY_IS_APPROX(m1.rint(), rint(m1));
Deanna Hood31fdd672015-03-11 06:39:23 +1000642 VERIFY_IS_APPROX(m1.floor(), floor(m1));
643 VERIFY_IS_APPROX(m1.ceil(), ceil(m1));
Christoph Hertzberg61e09772015-08-14 17:32:34 +0200644 VERIFY((m1.isNaN() == (Eigen::isnan)(m1)).all());
645 VERIFY((m1.isInf() == (Eigen::isinf)(m1)).all());
646 VERIFY((m1.isFinite() == (Eigen::isfinite)(m1)).all());
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800647 VERIFY_IS_APPROX(m4.inverse(), inverse(m4));
Deanna Hood41b717d2015-03-18 03:11:03 +1000648 VERIFY_IS_APPROX(m1.abs(), abs(m1));
649 VERIFY_IS_APPROX(m1.abs2(), abs2(m1));
650 VERIFY_IS_APPROX(m1.square(), square(m1));
651 VERIFY_IS_APPROX(m1.cube(), cube(m1));
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100652 VERIFY_IS_APPROX(cos(m1+RealScalar(3)*m2), cos((m1+RealScalar(3)*m2).eval()));
Gael Guennebaud3f32f5e2015-11-27 16:27:53 +0100653 VERIFY_IS_APPROX(m1.sign(), sign(m1));
Rasmus Munk Larsen6964ae82020-07-07 01:54:04 +0000654 VERIFY((m1.sqrt().sign().isNaN() == (Eigen::isnan)(sign(sqrt(m1)))).all());
Deanna Hood41b717d2015-03-18 03:11:03 +1000655
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800656 // avoid inf and NaNs so verification doesn't fail
657 m3 = m4.abs();
658 VERIFY_IS_APPROX(m3.sqrt(), sqrt(abs(m3)));
659 VERIFY_IS_APPROX(m3.rsqrt(), Scalar(1)/sqrt(abs(m3)));
660 VERIFY_IS_APPROX(rsqrt(m3), Scalar(1)/sqrt(abs(m3)));
Deanna Hood41b717d2015-03-18 03:11:03 +1000661 VERIFY_IS_APPROX(m3.log(), log(m3));
Gael Guennebaud89099b02016-06-01 17:00:08 +0200662 VERIFY_IS_APPROX(m3.log1p(), log1p(m3));
Deanna Hood41b717d2015-03-18 03:11:03 +1000663 VERIFY_IS_APPROX(m3.log10(), log10(m3));
Rasmus Munk Larsenf9fac1d2020-12-04 21:45:09 +0000664 VERIFY_IS_APPROX(m3.log2(), log2(m3));
Deanna Hood41b717d2015-03-18 03:11:03 +1000665
666
667 VERIFY((!(m1>m2) == (m1<=m2)).all());
668
669 VERIFY_IS_APPROX(sin(m1.asin()), m1);
670 VERIFY_IS_APPROX(cos(m1.acos()), m1);
671 VERIFY_IS_APPROX(tan(m1.atan()), m1);
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800672 VERIFY_IS_APPROX(sinh(m1), Scalar(0.5)*(exp(m1)-exp(-m1)));
673 VERIFY_IS_APPROX(cosh(m1), Scalar(0.5)*(exp(m1)+exp(-m1)));
674 VERIFY_IS_APPROX(tanh(m1), (Scalar(0.5)*(exp(m1)-exp(-m1)))/(Scalar(0.5)*(exp(m1)+exp(-m1))));
675 VERIFY_IS_APPROX(logistic(m1), (Scalar(1)/(Scalar(1)+exp(-m1))));
676 VERIFY_IS_APPROX(arg(m1), ((m1<Scalar(0)).template cast<Scalar>())*Scalar(std::acos(Scalar(-1))));
Deanna Hood41b717d2015-03-18 03:11:03 +1000677 VERIFY((round(m1) <= ceil(m1) && round(m1) >= floor(m1)).all());
Ilya Tokar19876ce2019-12-16 16:00:35 -0500678 VERIFY((rint(m1) <= ceil(m1) && rint(m1) >= floor(m1)).all());
679 VERIFY(((ceil(m1) - round(m1)) <= Scalar(0.5) || (round(m1) - floor(m1)) <= Scalar(0.5)).all());
680 VERIFY(((ceil(m1) - round(m1)) <= Scalar(1.0) && (round(m1) - floor(m1)) <= Scalar(1.0)).all());
681 VERIFY(((ceil(m1) - rint(m1)) <= Scalar(0.5) || (rint(m1) - floor(m1)) <= Scalar(0.5)).all());
682 VERIFY(((ceil(m1) - rint(m1)) <= Scalar(1.0) && (rint(m1) - floor(m1)) <= Scalar(1.0)).all());
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800683 VERIFY((Eigen::isnan)((m1*Scalar(0))/Scalar(0)).all());
684 VERIFY((Eigen::isinf)(m4/Scalar(0)).all());
685 VERIFY(((Eigen::isfinite)(m1) && (!(Eigen::isfinite)(m1*Scalar(0)/Scalar(0))) && (!(Eigen::isfinite)(m4/Scalar(0)))).all());
686 VERIFY_IS_APPROX(inverse(inverse(m4)),m4);
Deanna Hood41b717d2015-03-18 03:11:03 +1000687 VERIFY((abs(m1) == m1 || abs(m1) == -m1).all());
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800688 VERIFY_IS_APPROX(m3, sqrt(abs2(m3)));
Joel Holdsworthd5c66572020-03-19 17:45:20 +0000689 VERIFY_IS_APPROX(m1.absolute_difference(m2), (m1 > m2).select(m1 - m2, m2 - m1));
Gael Guennebaud3f32f5e2015-11-27 16:27:53 +0100690 VERIFY_IS_APPROX( m1.sign(), -(-m1).sign() );
691 VERIFY_IS_APPROX( m1*m1.sign(),m1.abs());
692 VERIFY_IS_APPROX(m1.sign() * m1.abs(), m1);
Rasmus Munk Larsen1e1848f2022-09-28 20:46:49 +0000693
694 ArrayType tmp = m1.atan2(m2);
695 for (Index i = 0; i < tmp.size(); ++i) {
696 Scalar actual = tmp.array()(i);
697 Scalar expected = atan2(m1.array()(i), m2.array()(i));
698 VERIFY_IS_APPROX(actual, expected);
699 }
Benoit Jacob5d63d2c2010-04-28 22:42:34 -0400700
Gael Guennebaud62670c82013-06-10 23:40:56 +0200701 VERIFY_IS_APPROX(numext::abs2(numext::real(m1)) + numext::abs2(numext::imag(m1)), numext::abs2(m1));
Gael Guennebaud87427d22019-10-08 09:15:17 +0200702 VERIFY_IS_APPROX(numext::abs2(Eigen::real(m1)) + numext::abs2(Eigen::imag(m1)), numext::abs2(m1));
Benoit Jacob5d63d2c2010-04-28 22:42:34 -0400703 if(!NumTraits<Scalar>::IsComplex)
Gael Guennebaud62670c82013-06-10 23:40:56 +0200704 VERIFY_IS_APPROX(numext::real(m1), m1);
Gael Guennebaud4ebb8042010-06-02 09:45:57 +0200705
Jitse Niesen6fecb6f2014-02-24 14:10:17 +0000706 // shift argument of logarithm so that it is not zero
707 Scalar smallNumber = NumTraits<Scalar>::dummy_precision();
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800708 VERIFY_IS_APPROX((m3 + smallNumber).log() , log(abs(m3) + smallNumber));
709 VERIFY_IS_APPROX((m3 + smallNumber + Scalar(1)).log() , log1p(abs(m3) + smallNumber));
Gael Guennebaud4ebb8042010-06-02 09:45:57 +0200710
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100711 VERIFY_IS_APPROX(m1.exp() * m2.exp(), exp(m1+m2));
712 VERIFY_IS_APPROX(m1.exp(), exp(m1));
713 VERIFY_IS_APPROX(m1.exp() / m2.exp(),(m1-m2).exp());
Gael Guennebaud575ac542010-06-19 23:17:07 +0200714
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800715 VERIFY_IS_APPROX(m1.expm1(), expm1(m1));
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800716 VERIFY_IS_APPROX((m3 + smallNumber).exp() - Scalar(1), expm1(abs(m3) + smallNumber));
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800717
Gael Guennebaud575ac542010-06-19 23:17:07 +0200718 VERIFY_IS_APPROX(m3.pow(RealScalar(0.5)), m3.sqrt());
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100719 VERIFY_IS_APPROX(pow(m3,RealScalar(0.5)), m3.sqrt());
Benoit Steinere25e3a02015-12-03 18:16:35 -0800720
721 VERIFY_IS_APPROX(m3.pow(RealScalar(-0.5)), m3.rsqrt());
722 VERIFY_IS_APPROX(pow(m3,RealScalar(-0.5)), m3.rsqrt());
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800723
724 // Avoid inf and NaN.
725 m3 = (m1.square()<NumTraits<Scalar>::epsilon()).select(Scalar(1),m3);
726 VERIFY_IS_APPROX(m3.pow(RealScalar(-2)), m3.square().inverse());
Rasmus Munk Larsen14c847d2022-10-12 20:12:08 +0000727
728 // Test pow and atan2 on special IEEE values.
729 binary_ops_test<Scalar>();
730 pow_scalar_exponent_test<Scalar>();
Benoit Steinere25e3a02015-12-03 18:16:35 -0800731
Antonio Sanchez4c42d5e2021-01-23 11:54:00 -0800732 VERIFY_IS_APPROX(log10(m3), log(m3)/numext::log(Scalar(10)));
733 VERIFY_IS_APPROX(log2(m3), log(m3)/numext::log(Scalar(2)));
Hauke Heibel81c13362012-03-07 08:58:42 +0100734
735 // scalar by array division
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100736 const RealScalar tiny = sqrt(std::numeric_limits<RealScalar>::epsilon());
Hauke Heibeldd9365e2012-03-09 14:04:13 +0100737 s1 += Scalar(tiny);
738 m1 += ArrayType::Constant(rows,cols,Scalar(tiny));
Rasmus Munk Larsen96dc37a2022-01-07 01:10:17 +0000739 VERIFY_IS_CWISE_APPROX(s1/m1, s1 * m1.inverse());
Eugene Brevdof7362772015-12-24 21:15:38 -0800740
Gael Guennebaudc695cbf2013-06-24 13:33:44 +0200741 // check inplace transpose
742 m3 = m1;
743 m3.transposeInPlace();
Deanna Hood41b717d2015-03-18 03:11:03 +1000744 VERIFY_IS_APPROX(m3, m1.transpose());
Gael Guennebaudc695cbf2013-06-24 13:33:44 +0200745 m3.transposeInPlace();
Deanna Hood41b717d2015-03-18 03:11:03 +1000746 VERIFY_IS_APPROX(m3, m1);
Gael Guennebaud0ce5bc02010-01-27 23:23:59 +0100747}
748
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +0000749
Jitse Niesen837db082011-05-09 10:17:41 +0100750template<typename ArrayType> void array_complex(const ArrayType& m)
751{
Deanna Hood41b717d2015-03-18 03:11:03 +1000752 typedef typename ArrayType::Scalar Scalar;
753 typedef typename NumTraits<Scalar>::Real RealScalar;
Jitse Niesen837db082011-05-09 10:17:41 +0100754
755 Index rows = m.rows();
756 Index cols = m.cols();
757
758 ArrayType m1 = ArrayType::Random(rows, cols),
Gael Guennebaud9fc1c922015-06-22 16:48:27 +0200759 m2(rows, cols),
760 m4 = m1;
Srinivas Vasudevan218764e2016-12-02 14:13:01 -0800761
Gael Guennebaud9fc1c922015-06-22 16:48:27 +0200762 m4.real() = (m4.real().abs()==RealScalar(0)).select(RealScalar(1),m4.real());
763 m4.imag() = (m4.imag().abs()==RealScalar(0)).select(RealScalar(1),m4.imag());
Jitse Niesen837db082011-05-09 10:17:41 +0100764
Deanna Hood41b717d2015-03-18 03:11:03 +1000765 Array<RealScalar, -1, -1> m3(rows, cols);
766
Jitse Niesen837db082011-05-09 10:17:41 +0100767 for (Index i = 0; i < m.rows(); ++i)
768 for (Index j = 0; j < m.cols(); ++j)
Gael Guennebauda76fbbf2012-11-06 15:25:50 +0100769 m2(i,j) = sqrt(m1(i,j));
Jitse Niesen837db082011-05-09 10:17:41 +0100770
Deanna Hood41b717d2015-03-18 03:11:03 +1000771 // these tests are mostly to check possible compilation issues with free-functions.
Deanna Hoodf89fcef2015-03-11 13:13:30 +1000772 VERIFY_IS_APPROX(m1.sin(), sin(m1));
773 VERIFY_IS_APPROX(m1.cos(), cos(m1));
774 VERIFY_IS_APPROX(m1.tan(), tan(m1));
775 VERIFY_IS_APPROX(m1.sinh(), sinh(m1));
776 VERIFY_IS_APPROX(m1.cosh(), cosh(m1));
777 VERIFY_IS_APPROX(m1.tanh(), tanh(m1));
Rasmus Munk Larsend6e283b2018-08-13 11:14:50 -0700778 VERIFY_IS_APPROX(m1.logistic(), logistic(m1));
Deanna Hood41b717d2015-03-18 03:11:03 +1000779 VERIFY_IS_APPROX(m1.arg(), arg(m1));
Christoph Hertzberg61e09772015-08-14 17:32:34 +0200780 VERIFY((m1.isNaN() == (Eigen::isnan)(m1)).all());
781 VERIFY((m1.isInf() == (Eigen::isinf)(m1)).all());
782 VERIFY((m1.isFinite() == (Eigen::isfinite)(m1)).all());
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800783 VERIFY_IS_APPROX(m4.inverse(), inverse(m4));
Deanna Hood41b717d2015-03-18 03:11:03 +1000784 VERIFY_IS_APPROX(m1.log(), log(m1));
785 VERIFY_IS_APPROX(m1.log10(), log10(m1));
Rasmus Munk Larsenf9fac1d2020-12-04 21:45:09 +0000786 VERIFY_IS_APPROX(m1.log2(), log2(m1));
Deanna Hood41b717d2015-03-18 03:11:03 +1000787 VERIFY_IS_APPROX(m1.abs(), abs(m1));
788 VERIFY_IS_APPROX(m1.abs2(), abs2(m1));
789 VERIFY_IS_APPROX(m1.sqrt(), sqrt(m1));
790 VERIFY_IS_APPROX(m1.square(), square(m1));
791 VERIFY_IS_APPROX(m1.cube(), cube(m1));
792 VERIFY_IS_APPROX(cos(m1+RealScalar(3)*m2), cos((m1+RealScalar(3)*m2).eval()));
Gael Guennebaud3f32f5e2015-11-27 16:27:53 +0100793 VERIFY_IS_APPROX(m1.sign(), sign(m1));
Deanna Hood41b717d2015-03-18 03:11:03 +1000794
Deanna Hood41b717d2015-03-18 03:11:03 +1000795 VERIFY_IS_APPROX(m1.exp() * m2.exp(), exp(m1+m2));
796 VERIFY_IS_APPROX(m1.exp(), exp(m1));
797 VERIFY_IS_APPROX(m1.exp() / m2.exp(),(m1-m2).exp());
798
Srinivas Vasudevan88062b72019-12-12 01:56:54 +0000799 VERIFY_IS_APPROX(m1.expm1(), expm1(m1));
800 VERIFY_IS_APPROX(expm1(m1), exp(m1) - 1.);
801 // Check for larger magnitude complex numbers that expm1 matches exp - 1.
802 VERIFY_IS_APPROX(expm1(10. * m1), exp(10. * m1) - 1.);
803
Deanna Hood41b717d2015-03-18 03:11:03 +1000804 VERIFY_IS_APPROX(sinh(m1), 0.5*(exp(m1)-exp(-m1)));
805 VERIFY_IS_APPROX(cosh(m1), 0.5*(exp(m1)+exp(-m1)));
806 VERIFY_IS_APPROX(tanh(m1), (0.5*(exp(m1)-exp(-m1)))/(0.5*(exp(m1)+exp(-m1))));
Rasmus Munk Larsend6e283b2018-08-13 11:14:50 -0700807 VERIFY_IS_APPROX(logistic(m1), (1.0/(1.0 + exp(-m1))));
Deanna Hood41b717d2015-03-18 03:11:03 +1000808
809 for (Index i = 0; i < m.rows(); ++i)
810 for (Index j = 0; j < m.cols(); ++j)
Gael Guennebaud87427d22019-10-08 09:15:17 +0200811 m3(i,j) = std::atan2(m1(i,j).imag(), m1(i,j).real());
Deanna Hood41b717d2015-03-18 03:11:03 +1000812 VERIFY_IS_APPROX(arg(m1), m3);
813
814 std::complex<RealScalar> zero(0.0,0.0);
Christoph Hertzberg61e09772015-08-14 17:32:34 +0200815 VERIFY((Eigen::isnan)(m1*zero/zero).all());
Gael Guennebaud4a985e72015-11-20 14:52:08 +0100816#if EIGEN_COMP_MSVC
817 // msvc complex division is not robust
818 VERIFY((Eigen::isinf)(m4/RealScalar(0)).all());
819#else
Gael Guennebaud40f326e2015-06-17 15:33:09 +0200820#if EIGEN_COMP_CLANG
Gael Guennebaud4a985e72015-11-20 14:52:08 +0100821 // clang's complex division is notoriously broken too
Christoph Hertzberg61e09772015-08-14 17:32:34 +0200822 if((numext::isinf)(m4(0,0)/RealScalar(0))) {
Gael Guennebaud40f326e2015-06-17 15:33:09 +0200823#endif
Gael Guennebaud4a985e72015-11-20 14:52:08 +0100824 VERIFY((Eigen::isinf)(m4/zero).all());
Gael Guennebaud40f326e2015-06-17 15:33:09 +0200825#if EIGEN_COMP_CLANG
826 }
827 else
828 {
Christoph Hertzberg61e09772015-08-14 17:32:34 +0200829 VERIFY((Eigen::isinf)(m4.real()/zero.real()).all());
Gael Guennebaud40f326e2015-06-17 15:33:09 +0200830 }
831#endif
Gael Guennebaud4a985e72015-11-20 14:52:08 +0100832#endif // MSVC
833
Christoph Hertzberg61e09772015-08-14 17:32:34 +0200834 VERIFY(((Eigen::isfinite)(m1) && (!(Eigen::isfinite)(m1*zero/zero)) && (!(Eigen::isfinite)(m1/zero))).all());
Deanna Hood41b717d2015-03-18 03:11:03 +1000835
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -0800836 VERIFY_IS_APPROX(inverse(inverse(m4)),m4);
Deanna Hood41b717d2015-03-18 03:11:03 +1000837 VERIFY_IS_APPROX(conj(m1.conjugate()), m1);
Gael Guennebaud87427d22019-10-08 09:15:17 +0200838 VERIFY_IS_APPROX(abs(m1), sqrt(square(m1.real())+square(m1.imag())));
Deanna Hood41b717d2015-03-18 03:11:03 +1000839 VERIFY_IS_APPROX(abs(m1), sqrt(abs2(m1)));
840 VERIFY_IS_APPROX(log10(m1), log(m1)/log(10));
Rasmus Munk Larsenf9fac1d2020-12-04 21:45:09 +0000841 VERIFY_IS_APPROX(log2(m1), log(m1)/log(2));
Deanna Hood41b717d2015-03-18 03:11:03 +1000842
Gael Guennebaud3f32f5e2015-11-27 16:27:53 +0100843 VERIFY_IS_APPROX( m1.sign(), -(-m1).sign() );
844 VERIFY_IS_APPROX( m1.sign() * m1.abs(), m1);
845
Deanna Hood41b717d2015-03-18 03:11:03 +1000846 // scalar by array division
Eugene Brevdof7362772015-12-24 21:15:38 -0800847 Scalar s1 = internal::random<Scalar>();
Benoit Steiner2d74ef92016-05-17 07:20:11 -0700848 const RealScalar tiny = std::sqrt(std::numeric_limits<RealScalar>::epsilon());
Deanna Hood41b717d2015-03-18 03:11:03 +1000849 s1 += Scalar(tiny);
850 m1 += ArrayType::Constant(rows,cols,Scalar(tiny));
851 VERIFY_IS_APPROX(s1/m1, s1 * m1.inverse());
852
853 // check inplace transpose
854 m2 = m1;
855 m2.transposeInPlace();
856 VERIFY_IS_APPROX(m2, m1.transpose());
857 m2.transposeInPlace();
858 VERIFY_IS_APPROX(m2, m1);
Rasmus Munk Larsenb47c7772020-04-27 18:55:15 -0700859 // Check vectorized inplace transpose.
Rasmus Munk Larsen74ec8e62020-05-07 17:29:56 +0000860 ArrayType m5 = ArrayType::Random(131, 131);
Rasmus Munk Larsenb47c7772020-04-27 18:55:15 -0700861 ArrayType m6 = m5;
862 m6.transposeInPlace();
863 VERIFY_IS_APPROX(m6, m5.transpose());
Jitse Niesen837db082011-05-09 10:17:41 +0100864}
865
Abraham Bachrach039408c2012-01-11 11:00:30 -0500866template<typename ArrayType> void min_max(const ArrayType& m)
867{
Abraham Bachrach039408c2012-01-11 11:00:30 -0500868 typedef typename ArrayType::Scalar Scalar;
869
870 Index rows = m.rows();
871 Index cols = m.cols();
872
873 ArrayType m1 = ArrayType::Random(rows, cols);
874
875 // min/max with array
876 Scalar maxM1 = m1.maxCoeff();
877 Scalar minM1 = m1.minCoeff();
878
879 VERIFY_IS_APPROX(ArrayType::Constant(rows,cols, minM1), (m1.min)(ArrayType::Constant(rows,cols, minM1)));
880 VERIFY_IS_APPROX(m1, (m1.min)(ArrayType::Constant(rows,cols, maxM1)));
881
882 VERIFY_IS_APPROX(ArrayType::Constant(rows,cols, maxM1), (m1.max)(ArrayType::Constant(rows,cols, maxM1)));
883 VERIFY_IS_APPROX(m1, (m1.max)(ArrayType::Constant(rows,cols, minM1)));
884
885 // min/max with scalar input
886 VERIFY_IS_APPROX(ArrayType::Constant(rows,cols, minM1), (m1.min)( minM1));
887 VERIFY_IS_APPROX(m1, (m1.min)( maxM1));
888
889 VERIFY_IS_APPROX(ArrayType::Constant(rows,cols, maxM1), (m1.max)( maxM1));
890 VERIFY_IS_APPROX(m1, (m1.max)( minM1));
891
Rasmus Munk Larsen841c8982021-02-24 17:49:20 -0800892
893 // min/max with various NaN propagation options.
894 if (m1.size() > 1 && !NumTraits<Scalar>::IsInteger) {
Antonio Sanchez8dfe1022021-03-16 20:12:46 -0700895 m1(0,0) = NumTraits<Scalar>::quiet_NaN();
Rasmus Munk Larsen841c8982021-02-24 17:49:20 -0800896 maxM1 = m1.template maxCoeff<PropagateNaN>();
897 minM1 = m1.template minCoeff<PropagateNaN>();
898 VERIFY((numext::isnan)(maxM1));
899 VERIFY((numext::isnan)(minM1));
900
901 maxM1 = m1.template maxCoeff<PropagateNumbers>();
902 minM1 = m1.template minCoeff<PropagateNumbers>();
903 VERIFY(!(numext::isnan)(maxM1));
904 VERIFY(!(numext::isnan)(minM1));
905 }
Abraham Bachrach039408c2012-01-11 11:00:30 -0500906}
907
Antonio Sanchezfc9d3522021-08-17 20:04:48 -0700908template<int N>
909struct shift_left {
910 template<typename Scalar>
911 Scalar operator()(const Scalar& v) const {
912 return v << N;
913 }
914};
915
916template<int N>
917struct arithmetic_shift_right {
918 template<typename Scalar>
919 Scalar operator()(const Scalar& v) const {
920 return v >> N;
921 }
922};
923
924template<typename ArrayType> void array_integer(const ArrayType& m)
925{
926 Index rows = m.rows();
927 Index cols = m.cols();
928
929 ArrayType m1 = ArrayType::Random(rows, cols),
930 m2(rows, cols);
931
932 m2 = m1.template shiftLeft<2>();
933 VERIFY( (m2 == m1.unaryExpr(shift_left<2>())).all() );
934 m2 = m1.template shiftLeft<9>();
935 VERIFY( (m2 == m1.unaryExpr(shift_left<9>())).all() );
936
937 m2 = m1.template shiftRight<2>();
938 VERIFY( (m2 == m1.unaryExpr(arithmetic_shift_right<2>())).all() );
939 m2 = m1.template shiftRight<9>();
940 VERIFY( (m2 == m1.unaryExpr(arithmetic_shift_right<9>())).all() );
941}
942
Charles Schlosser82b152d2022-11-04 00:31:20 +0000943template <typename ArrayType>
944struct signed_shift_test_impl {
945 typedef typename ArrayType::Scalar Scalar;
946 static constexpr size_t Size = sizeof(Scalar);
947 static constexpr size_t MaxShift = (CHAR_BIT * Size) - 1;
948
949 template <size_t N = 0>
950 static inline std::enable_if_t<(N > MaxShift), void> run(const ArrayType& ) {}
951 template <size_t N = 0>
952 static inline std::enable_if_t<(N <= MaxShift), void> run(const ArrayType& m) {
953 const Index rows = m.rows();
954 const Index cols = m.cols();
955
956 ArrayType m1 = ArrayType::Random(rows, cols), m2(rows, cols);
957
958 m2 = m1.unaryExpr([](const Scalar& x) { return x >> N; });
959 VERIFY((m2 == m1.unaryExpr(internal::scalar_shift_right_op<Scalar, N>())).all());
960
961 m2 = m1.unaryExpr([](const Scalar& x) { return x << N; });
962 VERIFY((m2 == m1.unaryExpr( internal::scalar_shift_left_op<Scalar, N>())).all());
963
964 run<N + 1>(m);
965 }
966};
967template <typename ArrayType>
968void signed_shift_test(const ArrayType& m) {
969 signed_shift_test_impl<ArrayType>::run(m);
970}
971
Gael Guennebaude0f6d352018-09-20 18:07:32 +0200972EIGEN_DECLARE_TEST(array_cwise)
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000973{
974 for(int i = 0; i < g_repeat; i++) {
Hauke Heibel2d0dfe52010-12-16 17:36:10 +0100975 CALL_SUBTEST_1( array(Array<float, 1, 1>()) );
Hauke Heibelb31e1242010-12-16 19:07:23 +0100976 CALL_SUBTEST_2( array(Array22f()) );
977 CALL_SUBTEST_3( array(Array44d()) );
Gael Guennebauda8f66fe2011-07-12 14:41:00 +0200978 CALL_SUBTEST_4( array(ArrayXXcf(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
979 CALL_SUBTEST_5( array(ArrayXXf(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
980 CALL_SUBTEST_6( array(ArrayXXi(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
Gael Guennebaud543529d2019-01-22 15:30:50 +0100981 CALL_SUBTEST_6( array(Array<Index,Dynamic,Dynamic>(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
Antonio Sanchezfc9d3522021-08-17 20:04:48 -0700982 CALL_SUBTEST_6( array_integer(ArrayXXi(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
983 CALL_SUBTEST_6( array_integer(Array<Index,Dynamic,Dynamic>(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
Charles Schlosser82b152d2022-11-04 00:31:20 +0000984 CALL_SUBTEST_7( signed_shift_test(ArrayXXi(internal::random<int>(1, EIGEN_TEST_MAX_SIZE), internal::random<int>(1, EIGEN_TEST_MAX_SIZE))));
985 CALL_SUBTEST_7( signed_shift_test(Array<Index, Dynamic, Dynamic>(internal::random<int>(1, EIGEN_TEST_MAX_SIZE), internal::random<int>(1, EIGEN_TEST_MAX_SIZE))));
986
Gael Guennebaud8cef5412008-06-21 17:28:07 +0000987 }
Hauke Heibel2d0dfe52010-12-16 17:36:10 +0100988 for(int i = 0; i < g_repeat; i++) {
989 CALL_SUBTEST_1( comparisons(Array<float, 1, 1>()) );
Hauke Heibelb31e1242010-12-16 19:07:23 +0100990 CALL_SUBTEST_2( comparisons(Array22f()) );
991 CALL_SUBTEST_3( comparisons(Array44d()) );
Gael Guennebauda8f66fe2011-07-12 14:41:00 +0200992 CALL_SUBTEST_5( comparisons(ArrayXXf(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
993 CALL_SUBTEST_6( comparisons(ArrayXXi(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
Hauke Heibel2d0dfe52010-12-16 17:36:10 +0100994 }
995 for(int i = 0; i < g_repeat; i++) {
Abraham Bachrach039408c2012-01-11 11:00:30 -0500996 CALL_SUBTEST_1( min_max(Array<float, 1, 1>()) );
997 CALL_SUBTEST_2( min_max(Array22f()) );
998 CALL_SUBTEST_3( min_max(Array44d()) );
999 CALL_SUBTEST_5( min_max(ArrayXXf(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
1000 CALL_SUBTEST_6( min_max(ArrayXXi(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
1001 }
1002 for(int i = 0; i < g_repeat; i++) {
Hauke Heibel2d0dfe52010-12-16 17:36:10 +01001003 CALL_SUBTEST_1( array_real(Array<float, 1, 1>()) );
Hauke Heibelb31e1242010-12-16 19:07:23 +01001004 CALL_SUBTEST_2( array_real(Array22f()) );
1005 CALL_SUBTEST_3( array_real(Array44d()) );
Gael Guennebauda8f66fe2011-07-12 14:41:00 +02001006 CALL_SUBTEST_5( array_real(ArrayXXf(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
Antonio Sanchezf0e46ed2021-01-22 11:10:54 -08001007 CALL_SUBTEST_7( array_real(Array<Eigen::half, 32, 32>()) );
1008 CALL_SUBTEST_8( array_real(Array<Eigen::bfloat16, 32, 32>()) );
Hauke Heibel2d0dfe52010-12-16 17:36:10 +01001009 }
Jitse Niesen837db082011-05-09 10:17:41 +01001010 for(int i = 0; i < g_repeat; i++) {
Gael Guennebauda8f66fe2011-07-12 14:41:00 +02001011 CALL_SUBTEST_4( array_complex(ArrayXXcf(internal::random<int>(1,EIGEN_TEST_MAX_SIZE), internal::random<int>(1,EIGEN_TEST_MAX_SIZE))) );
Jitse Niesen837db082011-05-09 10:17:41 +01001012 }
Benoit Jacob5d63d2c2010-04-28 22:42:34 -04001013
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +00001014 for(int i = 0; i < g_repeat; i++) {
1015 CALL_SUBTEST_6( int_pow_test() );
1016 CALL_SUBTEST_7( mixed_pow_test() );
Charles Schlosser82b152d2022-11-04 00:31:20 +00001017 CALL_SUBTEST_8( signbit_tests() );
Rasmus Munk Larsenafc014f2022-09-12 21:55:30 +00001018 }
1019
Hauke Heibel2d0dfe52010-12-16 17:36:10 +01001020 VERIFY((internal::is_same< internal::global_math_functions_filtering_base<int>::type, int >::value));
1021 VERIFY((internal::is_same< internal::global_math_functions_filtering_base<float>::type, float >::value));
1022 VERIFY((internal::is_same< internal::global_math_functions_filtering_base<Array2i>::type, ArrayBase<Array2i> >::value));
Gael Guennebaud64fcfd32016-06-14 11:26:57 +02001023 typedef CwiseUnaryOp<internal::scalar_abs_op<double>, ArrayXd > Xpr;
Hauke Heibel2d0dfe52010-12-16 17:36:10 +01001024 VERIFY((internal::is_same< internal::global_math_functions_filtering_base<Xpr>::type,
1025 ArrayBase<Xpr>
1026 >::value));
Gael Guennebaud8cef5412008-06-21 17:28:07 +00001027}