aboutsummaryrefslogtreecommitdiff
path: root/challenge-001/lubos-kolouch/python/ch-2.py
blob: 33cf5a935fc6fc4549cc28c541bb4b8bf54487b3 (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
32
33
34
35
36
37
38
39
40
41
#!/usr/bin/env python
# -*- coding: utf-8 -*-


def fizz_buzz():
    output = []
    for i in range(1, 21):
        line = ""
        if i % 3 == 0:
            line = "fizz"
        if i % 5 == 0:
            line += "buzz"
        if not line:
            line = i
        output.append(line)
    return output


# Simple test to verify the functionality
assert fizz_buzz() == [
    1,
    2,
    "fizz",
    4,
    "buzz",
    "fizz",
    7,
    8,
    "fizz",
    "buzz",
    11,
    "fizz",
    13,
    14,
    "fizzbuzz",
    16,
    17,
    "fizz",
    19,
    "buzz",
]