aboutsummaryrefslogtreecommitdiff
path: root/challenge-170/lubos-kolouch/python/ch-1.py
blob: e16744f2e75fbd14868c27c01d00f801c9a8a931 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
""" Challenge 170 Task 1 """
from sympy import nextprime


def gen_primorial_nums(count: int) -> list:
    """Generate the primorial numbers"""

    nums = [1]
    at_prime = 1

    while len(nums) < count:
        at_prime = nextprime(at_prime)

        nums.append(nums[-1] * at_prime)

    return nums


assert gen_primorial_nums(3) == [1, 2, 6]
assert gen_primorial_nums(10) == [
    1,
    2,
    6,
    30,
    210,
    2310,
    30030,
    510510,
    9699690,
    223092870,
]