python轻修（三）之自定义函数

通过自定义函数的形式算出1元二次方程的值

# Author:Liu Ren
#导入系统math函数
import math
def quadratic(a, b, c):
    d=b*b-4*a*c
    if d>=0:
        #通过方程来计算出一元二次方程的两个值
       sq=math.sqrt(d)
       zhi1=(-b+sq)/(2*a)
       zhi2=(-b-sq)/(2*a)
       yuanzu=(zhi1,zhi2)
       return yuanzu
    else:
       return -1

# 测试:
print('quadratic(2, 3, 1) =', quadratic(2, 3, 1))
print('quadratic(1, 3, -4) =', quadratic(1, 3, -4))
if quadratic(2, 3, 1) != (-0.5, -1.0):
    print('测试失败')
elif quadratic(1, 3, -4) != (1.0, -4.0):
    print('测试失败')
else:
    print('测试成功')