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
21template <class T>
22void
23test(const std::complex<T>& c, std::complex<T> x)
24{
25    assert(atan(c) == x);
26}
27
28template <class T>
29void
30test()
31{
32    test(std::complex<T>(0, 0), std::complex<T>(0, 0));
33}
34
35void 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
62int main()
63{
64    test<float>();
65    test<double>();
66    test<long double>();
67    test_edges();
68}
69