27 lines
760 B
C++
27 lines
760 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include <NBData/ComplexNumbers.hpp>
|
|
|
|
template<typename A, typename B>
|
|
void COMPLEX_EQ(const nb::Complex<A>& a, const nb::Complex<B>& b) {
|
|
ASSERT_EQ(a.r, b.r);
|
|
ASSERT_EQ(a.i, b.i);
|
|
}
|
|
|
|
template<typename A, typename B>
|
|
void COMPLEX_ALMOST_EQ(const nb::Complex<A>& a, const nb::Complex<B>& b) {
|
|
ASSERT_FLOAT_EQ(a.r, b.r);
|
|
ASSERT_FLOAT_EQ(a.i, b.i);
|
|
}
|
|
|
|
TEST(ComplexNumberTests, TestConstruction) {
|
|
COMPLEX_ALMOST_EQ(nb::Complex(), nb::Complex(0,0));
|
|
COMPLEX_ALMOST_EQ(nb::Complex(5.0), nb::Complex(5, 0));
|
|
COMPLEX_ALMOST_EQ(1.5_j, nb::Complex(0.0, 1.5));
|
|
}
|
|
|
|
TEST(ComplexNumberTests, TestAdd) {
|
|
auto x = nb::Complex(1.0) - nb::Complex(2.0, 1.0);
|
|
ASSERT_FLOAT_EQ(x.r, -1);
|
|
ASSERT_FLOAT_EQ(x.i, -1);
|
|
} |