PCC45 BNAndras

Author: BNAndras

45/BNAndras/fizzbuzz.py

@@ -0,0 +1,16 @@
+#!/usr/bin/env python3
+
+from typing import Union
+
+def fizzbuzz(n: int) -> Union[str, int]:
+    if not isinstance(n, int):
+        raise ValueError("Passed value is not a number.")
+
+    if n % 15 == 0:
+        return "FizzBuzz"
+    elif n % 5 == 0:
+        return "Buzz"
+    elif n % 3 == 0:
+        return "Fizz"
+    else:
+        return n

45/BNAndras/test_fizzbuzz.py

@@ -0,0 +1,24 @@
+from pytest import raises
+
+from fizzbuzz import fizzbuzz
+
+
+def test_raises_type_error_for_empty_arg():
+    with raises(TypeError):
+        fizzbuzz()
+
+def test_raises_value_error_for_non_number():
+    with raises(ValueError):
+        fizzbuzz("a")
+
+def test_returns_fizz():
+    assert fizzbuzz(3) == "Fizz"
+
+def test_returns_buzz():
+    assert fizzbuzz(5) == "Buzz"
+
+def test_returns_fizzbuzz():
+    assert fizzbuzz(15) == "FizzBuzz"
+
+def test_returns_value_given_number():
+    assert fizzbuzz(91) == 91