[scripts/pycalc] solve quadratic numbers with complex roots

script-resources/pycalc_startup.py

@@ -1,5 +1,7 @@
-from math import *
+import cmath
+import math
 from fractions import Fraction
+from math import *
 from typing import Set
 
 
@@ -13,20 +15,11 @@ def factors(n: int) -> Set[int]:
 
 
 def solve_quadratic(a: float, b: float, c: float) -> None:
-  if a == 0:
-    raise Exception("not a quadratic equation")
-  else:
-    d = b ** 2 - 4 * a * c
-    print("D = " + str(d))
-    if d < 0:
-      print("no solutions")
-    elif d > 0:
-      sd = sqrt(d)
-      print("sqrt(D) = " + str(sd))
-      print("x1 = " + str((-b + sd) / (2 * a)))
-      print("x2 = " + str((-b - sd) / (2 * a)))
-    else:
-      print("x = " + str(-b / (2 * a)))
+  d = b ** 2 - 4 * a * c
+  sd = cmath.sqrt(d)
+  print("sqrt(D) = " + str(sd))
+  print("x1 = " + str((-b + sd) / (2 * a)))
+  print("x2 = " + str((-b - sd) / (2 * a)))
 
 
 def cot(x: float) -> float: