Em Python, quando queremos escrever testes unitários podemos lançar mãos desde um simples assert, passando por funções auxiliares até um framework de testes unitários completo. Neste artigo vou mostrar rapidamente como funciona o statement assert e mostrar também algumas funções que também podem ser úteis, termino mostrando um "Hello world" do Unit testing framework.

assert

Em Python podemos fazer simples asserções utilizando o statement assert.

assert True

Se executarmos no terminal assert True, o comando fica em silêncio, nada responde. Esse é o comportamento esperado, pois a asserção teve sucesso e o "silêncio" é o sintoma de que tudo está funcionando.

Se executarmos assert False o comando dispara um erro...

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AssertionError

O statement aceita como segundo parâmetro uma string que pode ser um mensagem explicativa e é exibida quando a asserção falha.

>>> assert False, "mensagem explicativa"
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AssertionError: mensagem explicativa

Para exemplificar o funcionamento do statement, veja o trecho de código abaixo.

b = 150
expressao = 100 > b
mensagem  = "100 deve ser maior que 'b'"
assert expressao, mensagem

Obtemos como como resposta...

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AssertionError: 100 deve ser maior que b

O que vem depois ?

Se ao invés do "assert", você quisesse utilizar uma função, por exemplo, test() para realizar pequenos testes, como seria a implementação dessa função?

Que tal o exemplo abaixo?

def test(expression1, expression2):
    if expression1 == expression2:
        return 'Pass'
    else:
        return 'Fail'

Já o tutorial do Google sugere uma função igualmente simples, porém, ligeiramente mais sofisticada, veja código seguinte.

def test(got, expected):

  if got == expected:
    prefix = ' OK '
  else:
    prefix = '  X '

  print("%s got: %s expected: %s" % (prefix, repr(got), repr(expected)))

Pesquisando na web encontrei mais dois exemplos que merecem nossa atenção.

O primeiro deles, você confere no código abaixo. Repare que temos a utilização do objeto sys e que a saída dessa função é bem mais elaborada, ela alerta até o número da linha onde o erro foi encontrado.

import sys

def test(did_pass):
    """  Print the result of a test.  """
    linenum = sys._getframe(1).f_lineno   # Get the caller's line number.
    if did_pass:
        msg = "Test at line {0} ok.".format(linenum)
    else:
        msg = ("Test at line {0} FAILED.".format(linenum))
    print(msg)

O segundo exemplo é muito parecido com o anterior, até parece que o autor se baseou no primeiro exemplo e implementou sua própria customização. Pequenos detalhes foram alterados.

def test(actual, expected):
    import sys
    linenum = sys._getframe(1).f_lineno   # get the caller's line number.
    if (expected == actual):
        msg = "Test on line {0} passed.".format(linenum)
    else:
        msg = ("Test on line {0} failed. Expected '{1}', but got '{2}'."
                                     . format(linenum, expected, actual))
    print(msg)

Aí você percebe que precisa de um framework de verdade para escrever seus testes unitários...

Unit Testing Framework

O Unit Testing Framework pode ser o que você esteja procurando.

Longe de querer se aprofundar no framework (leia a documentação), vou apenas mostrar dois exemplos do tipo "Hello world".

Abaixo um exemplo demonstrando como testar uma simples função.

import unittest

def fun(x):
    return x + 1

class MyTest(unittest.TestCase):
    def test(self):
        self.assertEqual(fun(3), 4)

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

O resultado de python foo.py você confere abaixo.

.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

O exemplo seguinte demonstra como testar uma pequena classe.

import unittest

class MyFun:
    def fun(self, n):
        return n + 1

class MyFunTest(unittest.TestCase):
    def testFun(self):
        obj = MyFun()
        self.assertEqual(obj.fun(3), 4)

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

O resultado é semelhante ao anterior.

.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

Navegue nesta série!


Comentários

comments powered by Disqus