fizzbuzz_test.py

#!/usr/bin/env python

import unittest
from fizzbuzz import FizzBuzz

class TestFizzBuzz(unittest.TestCase):

	def test_numeroSimples(self):
		self.assertEqual(FizzBuzz(1), 1)
		self.assertEqual(FizzBuzz(2), 2)
		self.assertEqual(FizzBuzz(4), 4)

	def test_numeroMultiploDeTres(self):
		self.assertEqual(FizzBuzz(3), "fizz")
		self.assertEqual(FizzBuzz(9), "fizz")

	def test_numeroMultiploDeCinco(self):
		self.assertEqual(FizzBuzz(5), "buzz")

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