diff --git a/math/quadratic_equations_complex_numbers.cpp b/math/quadratic_equations_complex_numbers.cpp index 89296a698..a40c5c5ae 100644 --- a/math/quadratic_equations_complex_numbers.cpp +++ b/math/quadratic_equations_complex_numbers.cpp @@ -75,7 +75,21 @@ std::array, 2> quadraticEquation(long double a, return solutions; } +static void test() { + std::array, 2> realCase{ + std::complex{1, 0}, std::complex{1, 0}}; + assert(quadraticEquation(1, -2, 1) == realCase); + + std::array, 2> complexCase{ + std::complex{-2, -1}, std::complex{-2, 1}}; + assert(quadraticEquation(1, 4, 5) == complexCase); + + return; +} + int main() { + test(); + std::array, 2> aaa = quadraticEquation(1, 1, 1); std::cout << "The complex roots of x^2 + x + 1 are " << aaa[0] << " and " << aaa[1]