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 // <complex>
10 
11 // template<class T>
12 //   complex<T>
13 //   pow(const T& x, const complex<T>& y);
14 
15 #include <complex>
16 #include <cassert>
17 
18 #include "test_macros.h"
19 #include "../cases.h"
20 
21 template <class T>
22 void
23 test(const T& a, const std::complex<T>& b, std::complex<T> x)
24 {
25     std::complex<T> c = pow(a, b);
26     is_about(real(c), real(x));
27     assert(std::abs(imag(c)) < 1.e-6);
28 }
29 
30 template <class T>
31 void
32 test()
33 {
34     test(T(2), std::complex<T>(2), std::complex<T>(4));
35 }
36 
37 void test_edges()
38 {
39     const unsigned N = sizeof(testcases) / sizeof(testcases[0]);
40     for (unsigned i = 0; i < N; ++i)
41     {
42         for (unsigned j = 0; j < N; ++j)
43         {
44             std::complex<double> r = pow(real(testcases[i]), testcases[j]);
45             std::complex<double> z = exp(testcases[j] * log(std::complex<double>(real(testcases[i]))));
46             if (std::isnan(real(r)))
47                 assert(std::isnan(real(z)));
48             else
49             {
50                 assert(real(r) == real(z));
51             }
52             if (std::isnan(imag(r)))
53                 assert(std::isnan(imag(z)));
54             else
55             {
56                 assert(imag(r) == imag(z));
57             }
58         }
59     }
60 }
61 
62 int main(int, char**)
63 {
64     test<float>();
65     test<double>();
66     test<long double>();
67     test_edges();
68 
69   return 0;
70 }
71