diff --git a/fizzbuzz.py b/fizzbuzz.py new file mode 100644 index 0000000..a2da0e8 --- /dev/null +++ b/fizzbuzz.py @@ -0,0 +1,15 @@ +""" +fizzbuzz +""" + + +def FizzBuzz(n): + if isinstance(n, int): + if n % 15 == 0: + return "FizzBuzz" + elif n % 3 == 0: + return "Fizz" + elif n % 5 == 0: + return "Buzz" + else: + return n diff --git a/test_fizzbuzz.py b/test_fizzbuzz.py new file mode 100644 index 0000000..857f505 --- /dev/null +++ b/test_fizzbuzz.py @@ -0,0 +1,25 @@ +""" +test fizzbuzz + +""" + +import unittest +import fizzbuzz + + +class TestFizzBuzz(unittest.TestCase): + def test_num_3_and_5(self): + self.assertEqual(fizzbuzz.FizzBuzz(15), 'FizzBuzz') + + def test_num_3(self): + self.assertEqual(fizzbuzz.FizzBuzz(3), 'Fizz') + + def test_num_5(self): + self.assertEqual(fizzbuzz.FizzBuzz(5), 'Buzz') + + def test_num_other(self): + self.assertEqual(fizzbuzz.FizzBuzz(2), 2) + + +if __name__ == '__main__': + unittest.main()