diff --git a/src/prime_checker.py b/src/prime_checker.py new file mode 100644 index 00000000..10b2b01f --- /dev/null +++ b/src/prime_checker.py @@ -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 \ No newline at end of file diff --git a/tests/test_prime_checker.py b/tests/test_prime_checker.py new file mode 100644 index 00000000..7f941eab --- /dev/null +++ b/tests/test_prime_checker.py @@ -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" \ No newline at end of file