gistfile1.py

from unittest import TestCase
from mock import patch, Mock
from matriz import Matriz

class MatrizTest(TestCase):

    @patch('matriz.Matriz.inicializar_matriz')
    def test_intancia_matriz(self, method):
        method.return_value = 'matriz'
        matriz = Matriz(2, 2)
        self.assertEqual(2, matriz.linha)
        self.assertEqual(2, matriz.coluna)
        method.assert_called_once_with()
        self.assertEqual('matriz', matriz.dados)

    def test_inicializar_matriz_default(self):
        matriz = Matriz(2, 2)

        esperado = [
            [0,0],
            [0,0],
        ]

        resposta = matriz.inicializar_matriz()

        self.assertEqual(esperado, resposta)

    def test_inicializar_matriz_com_3(self):
        matriz = Matriz(4, 4)

        esperado = [
            [3, 3, 3],
            [3, 3, 3],
            [3, 3, 3],
        ]

        resposta = matriz.inicializar_matriz(3)

        self.assertEqual(esperado, resposta)

    def test_escreve_diagonal_default(self):
        matriz = Matriz(3, 3)
        matriz.escrever_diagonal()

        esperado = [
            [1, 0, 0],
            [0, 1, 0],
            [0, 0, 1],
        ]

        self.assertEqual(matriz.dados, esperado)

    def test_escreve_diagonal_com_4(self):
        matriz = Matriz(3, 3)
        matriz.escrever_diagonal(4)

        esperado = [
            [1, 0, 0],
            [0, 1, 0],
            [0, 0, 1],
        ]

        self.assertEqual(matriz.dados, esperado)

    def test_exception_nao_quadrada(self):
        with self.assertRaisesRegexp(Exception, 'matriz n quadrada$'):
            Matriz(2, 3)