Help me in solving PSPP78 problem

My issue

i didnt found answer in solution

My code


Learning course: Learn Python
Problem Link: CodeChef: Practical coding for everyone

Plzz refer the following solution .

# Review the code below and click on 'submit'

def calculate(a, b):
    result_1 = a**2 + b**3
    return result_1

for _ in range(3):
    A, B = map(int, input().split())
    result = calculate(A, B)
    print (result + result**2 + result**3)