From 7f8dc4f9e935d051482291085a7aa8e40f248dcc Mon Sep 17 00:00:00 2001 From: Caina Silva Date: Fri, 11 Jul 2025 18:29:15 -0700 Subject: [PATCH] Add is_prime function and an unit test for it) --- .../scipy_014/pr_tutorial/simple_functions.py | 8 ++++++++ .../pr_tutorial/tests/test_simple_function.py | 14 +++++++++++++- 2 files changed, 21 insertions(+), 1 deletion(-) diff --git a/participants/scipy_014/pr_tutorial/simple_functions.py b/participants/scipy_014/pr_tutorial/simple_functions.py index c3dcf8f..193a0a1 100644 --- a/participants/scipy_014/pr_tutorial/simple_functions.py +++ b/participants/scipy_014/pr_tutorial/simple_functions.py @@ -10,3 +10,11 @@ def factorial(value): return 1 else: return value * factorial(value - 1) + +def is_prime(value: int) -> bool: + if value <= 1: + return False + for i in range(2, value): + if value % i == 0: + return False + return True diff --git a/participants/scipy_014/pr_tutorial/tests/test_simple_function.py b/participants/scipy_014/pr_tutorial/tests/test_simple_function.py index 1982130..61f615e 100644 --- a/participants/scipy_014/pr_tutorial/tests/test_simple_function.py +++ b/participants/scipy_014/pr_tutorial/tests/test_simple_function.py @@ -1,7 +1,19 @@ -from pr_tutorial.simple_functions import factorial +from pr_tutorial.simple_functions import factorial, is_prime def test_factorial_3(): """Simplest test for one crete case""" assert factorial(3) == 6 + +def test_is_prime(): + """Test for is_prime function""" + assert not is_prime(1) + assert is_prime(2) + assert is_prime(3) + assert not is_prime(4) + assert is_prime(5) + assert not is_prime(6) + assert is_prime(7) + assert not is_prime(8) + assert not is_prime(9)