pythonは4桁の性質を判断する

【問題の説明】3025という数は独特の性質を持っている:それを二段、すなわち30と25に分けて、加算して平方を求める、すなわち(30+25)*(30+25)、ちょうど3025そのものに等しい.プログラミングプログラムは、入力された4桁nがabcd=(ab+cd)*(ab+cd)のような性質を満たす4桁であるか否かを判断し、1を出力し、そうでなければ0を出力する.【入力】4桁n【出力】1または0【サンプル入力】3025【サンプル出力】1【サンプル入力】3354【サンプル出力】0

n_str = input()
n = int(n_str)

n_left_half = int(n_str[:2])
n_right_half = int(n_str[2:])

if n == (n_left_half + n_right_half) ** 2:      >     
    print(1)
else:
    print(0)