| //===----------------------------------------------------------------------===// |
| // |
| // 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; |
| } |