Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 27 additions & 0 deletions src/prime_checker.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
def is_prime(number):
"""
Check if a given number is prime.

Args:
number (int): The number to check for primality.

Returns:
bool: True if the number is prime, False otherwise.

Raises:
ValueError: If the input is not a positive integer.
"""
# Check for invalid inputs
if not isinstance(number, int):
raise ValueError("Input must be an integer")

# Handle edge cases
if number < 2:
return False

# Check for divisibility up to the square root of the number
for i in range(2, int(number**0.5) + 1):
if number % i == 0:
return False

return True
41 changes: 41 additions & 0 deletions tests/test_prime_checker.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
import pytest
from src.prime_checker import is_prime

def test_prime_numbers():
"""Test known prime numbers."""
prime_numbers = [2, 3, 5, 7, 11, 13, 17, 19, 23, 29]
for num in prime_numbers:
assert is_prime(num) is True, f"{num} should be prime"

def test_non_prime_numbers():
"""Test known non-prime numbers."""
non_prime_numbers = [0, 1, 4, 6, 8, 9, 10, 12, 14, 15]
for num in non_prime_numbers:
assert is_prime(num) is False, f"{num} should not be prime"

def test_large_prime():
"""Test a larger prime number."""
assert is_prime(97) is True
assert is_prime(541) is True

def test_large_non_prime():
"""Test a larger non-prime number."""
assert is_prime(100) is False
assert is_prime(1000) is False

def test_invalid_inputs():
"""Test invalid input types."""
with pytest.raises(ValueError):
is_prime(3.14)

with pytest.raises(ValueError):
is_prime("not a number")

with pytest.raises(ValueError):
is_prime(None)

def test_negative_numbers():
"""Test negative numbers."""
negative_numbers = [-1, -2, -3, -17]
for num in negative_numbers:
assert is_prime(num) is False, f"{num} should not be prime"