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 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 30 test() 31 { 32 test(std::complex<T>(0, 0), std::complex<T>(0, 0)); 33 } 34 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 62 int main() 63 { 64 test<float>(); 65 test<double>(); 66 test<long double>(); 67 test_edges(); 68 } 69
