Ответ к предыдущей задаче для новичков

# Функция для вычисления округленного квадратного корня
def mySqrt(n):
if n == 0:
return 0
if n == 1:
return 1
for i in range(1, n + 1):
if i * i > n:
return i - 1

# Тесты для функции mySqrt()
def test_mySqrt():
assert mySqrt(0) == 0
assert mySqrt(4) == 2
assert mySqrt(9) == 3
assert mySqrt(16) == 4
assert mySqrt(17) == 4
assert mySqrt(25) == 5
assert mySqrt(26) == 5
assert mySqrt(30) == 5
assert mySqrt(31) == 5

#задача #coding