codeflash-internal/cli/code_to_optimize/final_test_set/generate_primes.py
2024-06-07 12:46:14 -07:00

21 lines
430 B
Python

def is_prime(n):
if n <= 1:
return False
if n <= 3:
return True
if n % 2 == 0 or n % 3 == 0:
return False
i = 5
while i * i <= n:
if n % i == 0 or n % (i + 2) == 0:
return False
i += 6
return True
def generate_primes(limit):
primes = []
for num in range(2, limit + 1):
if is_prime(num):
primes.append(num)
return primes