diff --git a/fb.py b/fb.py new file mode 100644 index 0000000..2f49627 --- /dev/null +++ b/fb.py @@ -0,0 +1,10 @@ +def FizzBuzz(i): + if i % 3 * i % 5: + return i + else: + return "Fizz"[i % 3 * 4:] + "Buzz"[i % 5 * 4:] + + +if __name__ == '__main__': + for i in range(1, 101): + print(FizzBuzz(i)) diff --git a/unit_TDD.py b/unit_TDD.py new file mode 100644 index 0000000..2ea9ea8 --- /dev/null +++ b/unit_TDD.py @@ -0,0 +1,14 @@ +import unittest +from fb import FizzBuzz + + +class Tf(unittest.TestCase): + def test_aa(self): + self.assertEquals(FizzBuzz(1), 1) + self.assertEquals(FizzBuzz(3), 'Fizz') + self.assertEquals(FizzBuzz(5), 'Buzz') + self.assertEquals(FizzBuzz(15), 'FizzBuzz') + + +if __name__ == '__main__': + unittest.main()