class FizzBuzzTest(unittest.TestCase):
def setUp(self):
self.numeros = list(range(101))
self.fizzBuzz = FizzBuzz()
def test_fizz(self):
"""Se o número for divisível por 3, diga Fizz"""
self.assertTrue(self.fizzBuzz.ehFizz(self.numeros[3]))
self.assertTrue(self.fizzBuzz.ehFizz(self.numeros[33]))
self.assertTrue(self.fizzBuzz.ehFizz(self.numeros[66]))
self.assertFalse(self.fizzBuzz.ehFizz(self.numeros[5]))
self.assertFalse(self.fizzBuzz.ehFizz(self.numeros[10]))
def test_buzz(self):
"""Se o número for divisível por 5, diga Buzz"""
self.assertTrue(self.fizzBuzz.ehBuzz(self.numeros[5]))
self.assertTrue(self.fizzBuzz.ehBuzz(self.numeros[55]))
self.assertTrue(self.fizzBuzz.ehBuzz(self.numeros[65]))
self.assertFalse(self.fizzBuzz.ehBuzz(self.numeros[3]))
self.assertFalse(self.fizzBuzz.ehBuzz(self.numeros[33]))
self.assertFalse(self.fizzBuzz.ehBuzz(self.numeros[93]))
def test_fizzBuzz(self):
"""Se o número for divisível por 3 e por 5, diga FizzBuzz"""
self.assertTrue(self.fizzBuzz.ehFizzBuzz(self.numeros[15]))
self.assertTrue(self.fizzBuzz.ehFizzBuzz(self.numeros[30]))
self.assertTrue(self.fizzBuzz.ehFizzBuzz(self.numeros[60]))
self.assertFalse(self.fizzBuzz.ehFizzBuzz(self.numeros[5]))
self.assertFalse(self.fizzBuzz.ehFizzBuzz(self.numeros[10]))
self.assertFalse(self.fizzBuzz.ehFizzBuzz(self.numeros[100]))
