Написать функцию is_square(n), определяющую, является ли указанное целое число полным квадратом. языка программирования python

#Python 3.6.1

def is_square(n):
   import math
   if n < 0:
      return False
   if math.sqrt(n) == int(math.sqrt(n)):
      return True
   else:
      return False
print("Введите целое число: ",end='')
if is_square(int(input())):
   print("Число является полным квадратом")
else:
   print("Число не является полным квадратом")