fizzbuzz.py

def fizzbuzz(N):
    vfb = ['Fizz', '', ''][N%3] + ['Buzz', '', '', '', ''][N%5]
    return vfb + (str(N)[len(vfb)*len(str(N)):])


import unittest
class TestFizzBuzz(unittest.TestCase):

    def test_fizzbuzz(self):
        for i in range(9999999):
            if i%3 == 0 and i%5 == 0:
                self.assertEqual(fizzbuzz(i), 'FizzBuzz')
            elif i%3 == 0:
                self.assertEqual(fizzbuzz(i), 'Fizz')
            elif i%5 == 0:
                self.assertEqual(fizzbuzz(i), 'Buzz')
            else:
                self.assertEqual(fizzbuzz(i), str(i))

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