From d71f851ecba42420346a987906e7d860aaa4d0d5 Mon Sep 17 00:00:00 2001 From: Muhammad Umer Farooq Date: Thu, 5 Oct 2023 22:21:43 +0500 Subject: [PATCH 1/2] Updated prime_numbers.py testcases. --- maths/prime_numbers.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/maths/prime_numbers.py b/maths/prime_numbers.py index c5297ed9264c..09a4fd2ba374 100644 --- a/maths/prime_numbers.py +++ b/maths/prime_numbers.py @@ -17,8 +17,8 @@ def slow_primes(max_n: int) -> Generator[int, None, None]: [2, 3, 5, 7, 11] >>> list(slow_primes(33)) [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31] - >>> list(slow_primes(10000))[-1] - 9973 + >>> list(slow_primes(1000))[-1] + 997 """ numbers: Generator = (i for i in range(1, (max_n + 1))) for i in (n for n in numbers if n > 1): @@ -44,8 +44,8 @@ def primes(max_n: int) -> Generator[int, None, None]: [2, 3, 5, 7, 11] >>> list(primes(33)) [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31] - >>> list(primes(10000))[-1] - 9973 + >>> list(primes(1000))[-1] + 997 """ numbers: Generator = (i for i in range(1, (max_n + 1))) for i in (n for n in numbers if n > 1): @@ -73,8 +73,8 @@ def fast_primes(max_n: int) -> Generator[int, None, None]: [2, 3, 5, 7, 11] >>> list(fast_primes(33)) [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31] - >>> list(fast_primes(10000))[-1] - 9973 + >>> list(fast_primes(1000))[-1] + 997 """ numbers: Generator = (i for i in range(1, (max_n + 1), 2)) # It's useless to test even numbers as they will not be prime @@ -97,9 +97,9 @@ def benchmark(): from timeit import timeit setup = "from __main__ import slow_primes, primes, fast_primes" - print(timeit("slow_primes(1_000_000_000_000)", setup=setup, number=1_000_000)) - print(timeit("primes(1_000_000_000_000)", setup=setup, number=1_000_000)) - print(timeit("fast_primes(1_000_000_000_000)", setup=setup, number=1_000_000)) + print(timeit("slow_primes(1_000)", setup=setup, number=1_000)) + print(timeit("primes(1_000)", setup=setup, number=1_000)) + print(timeit("fast_primes(1_000)", setup=setup, number=1_000)) if __name__ == "__main__": From 05c1300522f862842b8761df3900b6addc7ba376 Mon Sep 17 00:00:00 2001 From: Muhammad Umer Farooq Date: Thu, 5 Oct 2023 23:40:36 +0500 Subject: [PATCH 2/2] revert __main__ code. --- maths/prime_numbers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/maths/prime_numbers.py b/maths/prime_numbers.py index 09a4fd2ba374..38cc6670385d 100644 --- a/maths/prime_numbers.py +++ b/maths/prime_numbers.py @@ -97,9 +97,9 @@ def benchmark(): from timeit import timeit setup = "from __main__ import slow_primes, primes, fast_primes" - print(timeit("slow_primes(1_000)", setup=setup, number=1_000)) - print(timeit("primes(1_000)", setup=setup, number=1_000)) - print(timeit("fast_primes(1_000)", setup=setup, number=1_000)) + print(timeit("slow_primes(1_000_000_000_000)", setup=setup, number=1_000_000)) + print(timeit("primes(1_000_000_000_000)", setup=setup, number=1_000_000)) + print(timeit("fast_primes(1_000_000_000_000)", setup=setup, number=1_000_000)) if __name__ == "__main__":