matriz_test.py

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)
        dados = matriz.inicializar_matriz()
        self.assertEqual([[0, 0] ,[0, 0]], dados)

    def test_inicializar_matriz_com_3(self):
        matriz = Matriz(2, 2)
        dados = matriz.inicializar_matriz(3)
        self.assertEqual([[3 ,3], [3, 3]], dados)

    def test_escreve_diagonal_default(self):
        matriz = Matriz(2, 2)
        matriz.escrever_diagonal()
        self.assertEqual(1, matriz.dados[0][0])
        self.assertEqual(1, matriz.dados[1][1])

    def test_escreve_diagonal_com_4(self):
        matriz = Matriz(2, 2)
        matriz.escrever_diagonal(4)
        self.assertEqual(4, matriz.dados[0][0])
        self.assertEqual(4, matriz.dados[1][1])


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

Como conversamos, olha como fica mais legivel o código:
https://gist.github.com/edgabaldi/27f2f19c6400856486b6