Skip to content

Commit 358bc40

Browse files
authored
Refactor is_prime function
Replaced`for` loop in `is_prime` function with a generator expression.
1 parent 3b1f4c7 commit 358bc40

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

maths/emirp_check.py

+2-5
Original file line numberDiff line numberDiff line change
@@ -57,11 +57,8 @@ def is_prime(number: int) -> bool:
5757
elif number < 2 or (number % 2 == 0):
5858
return False
5959

60-
# Check divisibility from 3 to sqrt(number)
61-
for i in range(3, int(number**0.5) + 1, 2):
62-
if number % i == 0:
63-
return False
64-
return True
60+
# Return True if number is not divisible by any odd number up to its square root
61+
return all(number % i != 0 for i in range(3, int(number**0.5) + 1, 2))
6562

6663

6764
def emirp_check(number: int) -> bool:

0 commit comments

Comments
 (0)