test/std/numerics/complex.number/complex.transcendentals/atan.pass.cpp

   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 }