Lua Prime数字检查器

这是我的Lua代码,用于获取用户输入,并检查输入的数字是否为素数.我的问题是程序认为任何偶数都不是素数,而任何奇数都是.

print("Enter a number.")
 local number = io.read("*n")

 function prime(n)
 for i = 2, n^(1/2) do
   if (n % i) == 0 then
     return false
   end
   return true
 end
 end

 if prime(number) == true then
   print("Your number is prime!")
 end

 if prime(number) == false then
   print("Your number is not prime!")
 end

将返回true移出循环.

因此：

function prime(n)
    for i = 2, n^(1/2) do
        if (n % i) == 0 then
            return false
        end
    end
    return true
end