1.1 hw

# Написать функцию degree_of_two, которая определяет является ли заданное число квадратом какого-либо целого числа.
#
# Примеры:
# degree_of_two(81) ==> True
# degree_of_two(6561) ==> True
# degree_of_two(333)  ==> False

import traceback


def degree_of_two(n):
    # Тело функции
    i = float(n)
    i = n**0.5
    if i*i == n:
        return True
    else:
        return False

# Тесты
try:
    assert degree_of_two(1) == True
    assert degree_of_two(2) == False
    assert degree_of_two(5) == False
    assert degree_of_two(81) == True
    assert degree_of_two(6561) == True
except AssertionError:
    print("TEST ERROR")
    traceback.print_exc()
else:
    print("TEST PASSED")