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 //   atan(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(atan(c) == x);
26 }
27 
28 template <class T>
29 void
test()30 test()
31 {
32     test(std::complex<T>(0, 0), std::complex<T>(0, 0));
33 }
34 
test_edges()35 void test_edges()
36 {
37     typedef std::complex<double> C;
38     const unsigned N = sizeof(x) / sizeof(x[0]);
39     for (unsigned i = 0; i < N; ++i)
40     {
41         std::complex<double> r = atan(x[i]);
42         std::complex<double> t1(-imag(x[i]), real(x[i]));
43         std::complex<double> t2 = atanh(t1);
44         std::complex<double> z(imag(t2), -real(t2));
45         if (std::isnan(real(r)))
46             assert(std::isnan(real(z)));
47         else
48         {
49             assert(real(r) == real(z));
50             assert(std::signbit(real(r)) == std::signbit(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             assert(std::signbit(imag(r)) == std::signbit(imag(z)));
58         }
59     }
60 }
61 
main()62 int main()
63 {
64     test<float>();
65     test<double>();
66     test<long double>();
67     test_edges();
68 }
69