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 // <complex>
11
12 // template<class T>
13 // complex<T>
14 // log(const complex<T>& x);
15
16 #include <complex>
17 #include <cassert>
18
19 #include "../cases.h"
20
21 template <class T>
22 void
test(const std::complex<T> & c,std::complex<T> x)23 test(const std::complex<T>& c, std::complex<T> x)
24 {
25 assert(log(c) == x);
26 }
27
28 template <class T>
29 void
test()30 test()
31 {
32 test(std::complex<T>(0, 0), std::complex<T>(-INFINITY, 0));
33 }
34
test_edges()35 void test_edges()
36 {
37 const double pi = std::atan2(+0., -0.);
38 const unsigned N = sizeof(x) / sizeof(x[0]);
39 for (unsigned i = 0; i < N; ++i)
40 {
41 std::complex<double> r = log(x[i]);
42 if (x[i].real() == 0 && x[i].imag() == 0)
43 {
44 if (std::signbit(x[i].real()))
45 {
46 assert(std::isinf(r.real()));
47 assert(r.real() < 0);
48 if (std::signbit(x[i].imag()))
49 is_about(r.imag(), -pi);
50 else
51 is_about(r.imag(), pi);
52 }
53 else
54 {
55 assert(std::isinf(r.real()));
56 assert(r.real() < 0);
57 assert(r.imag() == 0);
58 assert(std::signbit(x[i].imag()) == std::signbit(r.imag()));
59 }
60 }
61 else if (std::isfinite(x[i].real()) && std::isinf(x[i].imag()))
62 {
63 assert(std::isinf(r.real()));
64 assert(r.real() > 0);
65 if (x[i].imag() > 0)
66 is_about(r.imag(), pi/2);
67 else
68 is_about(r.imag(), -pi/2);
69 }
70 else if (std::isfinite(x[i].real()) && std::isnan(x[i].imag()))
71 {
72 assert(std::isnan(r.real()));
73 assert(std::isnan(r.imag()));
74 }
75 else if (std::isinf(x[i].real()) && x[i].real() < 0 && std::isfinite(x[i].imag()))
76 {
77 assert(std::isinf(r.real()) && r.real() > 0);
78 if (r.imag() > 0)
79 is_about(r.imag(), pi);
80 else
81 is_about(r.imag(), -pi);
82 }
83 else if (std::isinf(x[i].real()) && x[i].real() > 0 && std::isfinite(x[i].imag()))
84 {
85 assert(std::isinf(r.real()) && r.real() > 0);
86 assert(r.imag() == 0);
87 assert(std::signbit(x[i].imag()) == std::signbit(r.imag()));
88 }
89 else if (x[i].real() == 1 && x[i].imag() == 0)
90 {
91 assert(r.real() == 0);
92 assert(std::signbit(r.imag()) == std::signbit(x[i].imag()));
93 }
94 else if (x[i].real() == 0 && x[i].imag() == 1)
95 {
96 assert(r.real() == 0);
97 is_about(r.imag(), pi/2);
98 }
99 else if (x[i].real() == -1 && x[i].imag() == 0)
100 {
101 assert(r.real() == 0);
102 if (std::signbit(x[i].imag()))
103 is_about(r.imag(), -pi);
104 else
105 is_about(r.imag(), pi);
106 }
107 else if (x[i].real() == 0 && x[i].imag() == -1)
108 {
109 assert(r.real() == 0);
110 is_about(r.imag(), -pi/2);
111 }
112 else if (std::isfinite(x[i].real()) && std::isfinite(x[i].imag()) && abs(x[i]) < 1)
113 {
114 assert( std::signbit(r.real()));
115 assert(std::signbit(r.imag()) == std::signbit(x[i].imag()));
116 }
117 else if (std::isfinite(x[i].real()) && std::isfinite(x[i].imag()) && abs(x[i]) > 1)
118 {
119 assert(!std::signbit(r.real()));
120 assert(std::signbit(r.imag()) == std::signbit(x[i].imag()));
121 }
122 }
123 }
124
main()125 int main()
126 {
127 test<float>();
128 test<double>();
129 test<long double>();
130 test_edges();
131 }
132