Python Exemplo Teste Unitário

calculadora.py

# -*- coding: utf-8 -*-
"""Módulo da Calculadora."""


class Calculadora:
    """Classe da Calculadora."""

    def soma(self, n1, n2):
        """Metodo que soma dois números."""
        return n1 + n2

    def subtracao(self, n1, n2):
        """Metodo que subtrai o n1 pelo n2."""
        return n1 - n2

    def multiplicacao(self, n1, n2):
        """Metodo que multiplica dois números."""
        return n1 * n2

    def divisao(self, n1, n2):
        """Metodo que divide n1 por n2."""
        return n1 / n2

calculadora_test.py

# -*- coding: utf-8 -*-
"""Módulo de Testes."""
import unittest
from calculadora import Calculadora


class CalculadoraTestCase(unittest.TestCase):
    """Classe para testar os paranaues."""

    def setUp(self):
        """Inicializando Teste."""
        self.calculadora = Calculadora()

    def test_soma(self):
        """Metodo testando soma."""
        self.assertEqual(self.calculadora.soma(2, 2), 4)
        self.assertEqual(self.calculadora.soma(1, 5), 6)
        self.assertEqual(self.calculadora.soma(5, -3), 2)
        self.assertEqual(self.calculadora.soma(5, 0), 5)

    def test_subtracao(self):
        """Metodo testando subtracao."""
        self.assertEqual(self.calculadora.subtracao(2, 2), 0)
        self.assertEqual(self.calculadora.subtracao(1, 5), -4)
        self.assertEqual(self.calculadora.subtracao(5, -3), 8)
        self.assertEqual(self.calculadora.subtracao(5, 0), 5)

    def test_multiplicacao(self):
        """Metodo testando multiplicacao."""
        self.assertEqual(self.calculadora.multiplicacao(2, 2), 4)
        self.assertEqual(self.calculadora.multiplicacao(1, 5), 5)
        self.assertEqual(self.calculadora.multiplicacao(5, -3), -15)
        self.assertEqual(self.calculadora.multiplicacao(5, 0), 0)

    def test_divisao(self):
        """Metodo testando divisao."""
        self.assertEqual(self.calculadora.divisao(2, 2), 1)
        self.assertEqual(self.calculadora.divisao(1, 5), 0.2)
        with self.assertRaises(ZeroDivisionError):
            self.assertEqual(self.calculadora.divisao(5, 0), 0)


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