def test_rain_nao_retornando_erro_de_KeyboardInterrupt_2(self, mock, *_): matrix = MagicMock() mock.return_value = matrix matrix.condicoes.side_effect = [True] * 3 + [False] matrix.rain.side_effect = [KeyboardInterrupt()] * 3 + [0] main('') self.assertEqual(matrix.condicoes.call_count, 4)
def test_texto_efeito_pausa_chamado_com_argumentos(self, mock2, *_): main('') esperado = [ call('Conectando a matrix...'), call(attr(0) + '\nDesconectado.') ] self.assertEqual(mock2.mock_calls, esperado)
def test_rain_nao_retornando_erro_de_KeyboardInterrupt(self, mock, *_): mock.side_effect = [KeyboardInterrupt()] main('') self.assertEqual(mock.call_count, 1)
def test_rain_chamado(self, mock, *_): main('') mock.assert_any_call()
def test_arquiteto_chamado(self, mock, *_): main('') self.assertEqual(mock.call_count, 1)
from sys import argv from matrix_.matrix import main if __name__ == '__main__': main(' '.join(argv[1:]))