Help me in solving PSPP78 problem

help
,
1

My issue

i didnt found answer in solution

My code

8

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

2

@nishma8351
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)