Python Fizzbuzz

July 15, 2018

Test file

Create file fizz_buzz_test.py:

import unittest
import fizz_buzz


class FizzBuzzTest(unittest.TestCase):
    def test_capitalise_sentence(self):
        assess = [
            {
                "input": 2,
                "expectation": 2
            },
            {
                "input": 3,
                "expectation": "Fizz"
            },
            {
                "input": 5,
                "expectation": "Buzz"
            },
            {
                "input": 15,
                "expectation": "FizzBuzz"
            },
        ]
        for test in assess:
            self.assertEqual(fizz_buzz.run(
                test["input"]), test["expectation"])


if __name__ == '__main__':
    unittest.main()

FizzBuzz

Create file fizz_buzz.py.

def run(arg):
    if arg % 5 == 0 and arg % 3 == 0:
        return "FizzBuzz"
    elif arg % 3 == 0:
        return "Fizz"
    elif arg % 5 == 0:
        return "Buzz"
    else:
        return arg

Running tests

Change into directory and run python3 -m pytest -v fizz_buzz_test.py.

Related Articles

• ← Ruby Spiral Matrix
• PHP Spiral Matrix →