Skip to content

Commit 40d7e7a

Browse files
author
sailok.chinta
committed
feat: fix test cases
1 parent 74d6ab5 commit 40d7e7a

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

src/main/java/com/thealgorithms/maths/QuadraticEquationSolver.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
package com.thealgorithms.maths;
22

3+
import java.util.Arrays;
4+
35
class ComplexNumber {
46
Double real;
57
Double imaginary;
@@ -48,7 +50,7 @@ public ComplexNumber[] solveEquation(double a, double b, double c) {
4850
double realPart = -b / (2 * a);
4951
double imaginaryPart = Math.sqrt(-discriminant) / (2 * a);
5052

51-
return new ComplexNumber[] {new ComplexNumber(realPart, imaginaryPart), new ComplexNumber(realPart, imaginaryPart)};
53+
return new ComplexNumber[] {new ComplexNumber(realPart, imaginaryPart), new ComplexNumber(realPart, -imaginaryPart)};
5254
}
5355

5456
// return no roots

src/test/java/com/thealgorithms/maths/QuadraticEquationSolverTest.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,10 @@ public void testSolveEquation_realRoots() {
1515

1616
ComplexNumber[] roots = quadraticEquationSolver.solveEquation(a, b, c);
1717
Assertions.assertEquals(roots.length, 2);
18-
Assertions.assertEquals(roots[0].real, -1.6266572504051);
19-
Assertions.assertEquals(roots[1].real, -0.27810465435684306);
18+
Assertions.assertEquals(roots[0].real, -0.27810465435684306);
19+
Assertions.assertNull(roots[0].imaginary);
20+
Assertions.assertEquals(roots[1].real, -1.6266572504050616);
21+
Assertions.assertNull(roots[1].imaginary);
2022
}
2123

2224
@Test

0 commit comments

Comments
 (0)