diff --git a/FizzBuzz.py b/FizzBuzz.py new file mode 100644 index 0000000..afbf415 --- /dev/null +++ b/FizzBuzz.py @@ -0,0 +1,10 @@ +def transform(num): + ret = '' + if num % 3 == 0: + ret += 'Fizz' + if num % 5 == 0: + ret += 'Buzz' + if ret: + return ret + else: + return str(num) diff --git a/Test_FizzBuzz.py b/Test_FizzBuzz.py new file mode 100644 index 0000000..b6b5847 --- /dev/null +++ b/Test_FizzBuzz.py @@ -0,0 +1,20 @@ +from FizzBuzz import transform +import unittest + + +class TestFizzBuzz(unittest.TestCase): + def test_nomal(self): + self.assertEqual(transform(2), '2') + + def test_times_3(self): + self.assertEqual(transform(3), 'Fizz') + + def test_times_5(self): + self.assertEqual(transform(5), 'Buzz') + + def test_times_3_5(self): + self.assertEqual(transform(15), 'FizzBuzz') + + +if __name__ == '__main__': + unittest.main()