python编译用自定义函数的方法

python编写一个程序，用自定义函数的方法，求输入数字的平方和。

望采纳！谢谢

def cal(n):
  return n**2

print(cal(int(input())))