結果 : square in python using math