41번 문제 질문입니다.

def chech_prime(n):

if n <= 1:

return "NO"

i = 2

소수 = True

while (i**2) < n:

if n % i == 0:

소수 = False

break

i += 1

if 소수:

return "YES"

else:

return "NO"

print(chech_prime(4))

4를 넣었을 때 'NO'가 아니라 'Yes'가 나옵니다.

 while (i**2) <= n: 여기에 등호를 추가해야 하는거 아닌가요?