Вычислить значение функции

import math
b = 1.25
x = float(raw_input("input x $ "))
if x<1:
    y = x*x - 2
elif x==1:
    y = math.tan(3.14/4)*math.tan(3.14/4)+x
elif x>1:
    y = math.sqrt(b*b*b+x*x)
 
print("Result: ")
print y
raw_input("\nPRESS ANY KEY FOR EXIT")

import math
 
b = 1.25
 
x = float(raw_input("x $ "))
 
if x < 1:
    y = x**2 - 2
elif x == 1:
    y = math.tan(math.pi/4) + x
elif x > 1:
    y = math.sqrt(b**3 + x**2)
 
print y

def y(b):
    def foo(x):
        if x < 1:
            return x**2 - 2
        elif x == 1:
            return 1 + x
        else:
            return math.sqrt(b**3 + x**2)
    return foo
 
# test
 
>>> y1 = y(1.25)
>>> for a in map(y1, [0.0, 0.5, 1.0, 1.5, 3.0]):
        print(a)
    
-2.0
-1.75
2.0
2.050152433357091
3.3095505737184316
>>>

return (b ** 3 + x ** 2) ** 0.5