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)