1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
// <complex>
// template<class T>
// complex<T>
// sqrt(const complex<T>& x);
#include <complex>
#include <cassert>
#include "test_macros.h"
#include "../cases.h"
template <class T>
void
test(const std::complex<T>& c, std::complex<T> x)
{
std::complex<T> a = sqrt(c);
is_about(real(a), real(x));
assert(std::abs(imag(c)) < 1.e-6);
}
template <class T>
void
test()
{
test(std::complex<T>(64, 0), std::complex<T>(8, 0));
}
void test_edges()
{
const unsigned N = sizeof(testcases) / sizeof(testcases[0]);
for (unsigned i = 0; i < N; ++i)
{
std::complex<double> r = sqrt(testcases[i]);
if (testcases[i].real() == 0 && testcases[i].imag() == 0)
{
assert(!std::signbit(r.real()));
assert(std::signbit(r.imag()) == std::signbit(testcases[i].imag()));
}
else if (std::isinf(testcases[i].imag()))
{
assert(std::isinf(r.real()));
assert(r.real() > 0);
assert(std::isinf(r.imag()));
assert(std::signbit(r.imag()) == std::signbit(testcases[i].imag()));
}
else if (std::isfinite(testcases[i].real()) && std::isnan(testcases[i].imag()))
{
assert(std::isnan(r.real()));
assert(std::isnan(r.imag()));
}
else if (std::isinf(testcases[i].real()) && testcases[i].real() < 0 && std::isfinite(testcases[i].imag()))
{
assert(r.real() == 0);
assert(!std::signbit(r.real()));
assert(std::isinf(r.imag()));
assert(std::signbit(testcases[i].imag()) == std::signbit(r.imag()));
}
else if (std::isinf(testcases[i].real()) && testcases[i].real() > 0 && std::isfinite(testcases[i].imag()))
{
assert(std::isinf(r.real()));
assert(r.real() > 0);
assert(r.imag() == 0);
assert(std::signbit(testcases[i].imag()) == std::signbit(r.imag()));
}
else if (std::isinf(testcases[i].real()) && testcases[i].real() < 0 && std::isnan(testcases[i].imag()))
{
assert(std::isnan(r.real()));
assert(std::isinf(r.imag()));
}
else if (std::isinf(testcases[i].real()) && testcases[i].real() > 0 && std::isnan(testcases[i].imag()))
{
assert(std::isinf(r.real()));
assert(r.real() > 0);
assert(std::isnan(r.imag()));
}
else if (std::isnan(testcases[i].real()) && (std::isfinite(testcases[i].imag()) || std::isnan(testcases[i].imag())))
{
assert(std::isnan(r.real()));
assert(std::isnan(r.imag()));
}
else if (std::signbit(testcases[i].imag()))
{
assert(!std::signbit(r.real()));
assert(std::signbit(r.imag()));
}
else
{
assert(!std::signbit(r.real()));
assert(!std::signbit(r.imag()));
}
}
}
int main(int, char**)
{
test<float>();
test<double>();
test<long double>();
test_edges();
return 0;
}
|
