dojo_11_01_2013.py

import unittest

def fizzbuzz(numero):
    resultado = ""
    if numero%3 == 0:
        resultado += 'fizz'
    if numero%5 == 0:
        resultado += 'buzz'
    if resultado == "":
        resultado = numero
    return resultado 
 
def listfizz (lista):
    lista_saida=list()
    for elemento in lista:
        lista_saida.append(fizzbuzz(elemento))
    return lista_saida
    
class Test(unittest.TestCase):

    def test_1_tem_que_retornar_1(self):
        resultado = fizzbuzz(1)
        self.assertEquals(resultado, 1)

    def test_2_tem_que_retornar_2(self):
        resultado = fizzbuzz(2)
        self.assertEquals(resultado, 2)
        
    def test_3_tem_que_retornar_fizz(self):
        resultado = fizzbuzz(3)
        self.assertEquals(resultado, 'fizz')
    
    def test_5_tem_que_retornar_buzz(self):
        resultado = fizzbuzz(5)
        self.assertEquals(resultado, 'buzz')
    
    def test_15_tem_que_retornar_fizzbuzz(self):
        resultado = fizzbuzz(15)
        self.assertEquals(resultado, 'fizzbuzz')

    def test_lista_vazia(self):
        resultado = listfizz([])
        self.assertEquals(resultado, [])
    
    def test_lista_um_elemento(self):
        resultado= listfizz([1])
        self.assertEquals(resultado, [1])
        
    def test_lista_tres_elemento(self):
        resultado= listfizz([1,2,3])
        self.assertEquals(resultado, [1,2,'fizz'])


unittest.main()