From 685469d2c45e2f9de13ddefdc7791967094fcbe1 Mon Sep 17 00:00:00 2001
From: Gahjouyooj <tyiwei0@gmail.com>
Date: Wed, 19 Apr 2023 19:45:29 +0800
Subject: [PATCH] Added test cases

---
 math/quadratic_equations_complex_numbers.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

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<std::complex<long double>, 2> quadraticEquation(long double a,
     return solutions;
 }
 
+static void test() {
+    std::array<std::complex<long double>, 2> realCase{
+        std::complex<long double>{1, 0}, std::complex<long double>{1, 0}};
+    assert(quadraticEquation(1, -2, 1) == realCase);
+
+    std::array<std::complex<long double>, 2> complexCase{
+        std::complex<long double>{-2, -1}, std::complex<long double>{-2, 1}};
+    assert(quadraticEquation(1, 4, 5) == complexCase);
+
+    return;
+}
+
 int main() {
+    test();
+
     std::array<std::complex<long double>, 2> aaa = quadraticEquation(1, 1, 1);
     std::cout << "The complex roots of x^2 + x + 1 are " << aaa[0] << " and "
               << aaa[1]