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 const unsigned N = sizeof(testcases) / sizeof(testcases[0]); 38 for (unsigned i = 0; i < N; ++i) 39 { 40 std::complex<double> r = atan(testcases[i]); 41 std::complex<double> t1(-imag(testcases[i]), real(testcases[i])); 42 std::complex<double> t2 = atanh(t1); 43 std::complex<double> z(imag(t2), -real(t2)); 44 if (std::isnan(real(r))) 45 assert(std::isnan(real(z))); 46 else 47 { 48 assert(real(r) == real(z)); 49 assert(std::signbit(real(r)) == std::signbit(real(z))); 50 } 51 if (std::isnan(imag(r))) 52 assert(std::isnan(imag(z))); 53 else 54 { 55 assert(imag(r) == imag(z)); 56 assert(std::signbit(imag(r)) == std::signbit(imag(z))); 57 } 58 } 59 } 60 61 int main() 62 { 63 test<float>(); 64 test<double>(); 65 test<long double>(); 66 test_edges(); 67 } 68
