結果 : check if square python