pocketlang/tests/examples/prime.pk

26 lines
373 B
Plaintext
Raw Normal View History

## Prime numbers.
2021-05-13 18:31:55 +08:00
def is_prime(n)
if n < 2 then return false end
for i in 2..n
if n % i == 0 then return false end
end
return true
end
def get_all_primes(n)
res = []
2021-05-13 18:31:55 +08:00
for i in 0..n
if is_prime(i)
list_append(res, i)
end
2021-05-13 18:31:55 +08:00
end
return res
2021-05-13 18:31:55 +08:00
end
primes = get_all_primes(20)
assert(primes == [2, 3, 5, 7, 11, 13, 17, 19])
print('primes =', primes)