dotfiles/script-resources/pycalc_startup.py

from math import *
from fractions import Fraction


def factors(n):
  result = set()
  for i in range(1, int(sqrt(n)) + 1):
    if n % i == 0:
      result.add(i)
      result.add(n // i)
  return result


def solve_quadratic(a, b, c):
  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)))


print("loaded Python calculator")