어떤 자연수를 제곱했을 때 나오는 정수를 제곱수라고 합니다. 정수 n이 매개변수로 주어질 때, n이 제곱수라면 1을 아니라면 2를 return하도록 solution 함수를 완성해주세요.

def solution(n):
tmp = n ** (1/2)
if round(tmp,1)==tmp:
answer = 1
else:
answer=2
return answer
def solution(n):
return 1 if (n ** 0.5).is_integer() else 2

is_integer() 함수
# float 객체가 유한한 정수이면 True, 그렇지 않으면 False 반환
# Syntax
float.is_integer()
# example
1.5.is_integer() # False
1.0.is_integer() # True