def test_matriz_presencas(self): MATRIZ_PRESENCAS_ESPERADA = numpy.matrix([ [1., 1., 1., 1., 1., 1., 1., 1.], [1., 1., 1., 1., 1., 1., 1., 1.], [1., 1., 1., 1., 1., 1., 1., 1.], [1., 1., 1., 1., 1., 1., 1., 1.], [1., 1., 1., 1., 1., 1., 1., 1.], [1., 1., 1., 1., 1., 1., 1., 1.], [1., 1., 1., 1., 1., 0., 1., 1.], [1., 1., 1., 1., 1., 1., 0., 0.], [1., 1., 1., 0., 1., 1., 1., 1.] ]) builder = analise.MatrizesDeDadosBuilder(self.votacoes, self.partidos, self.legislaturas) builder.gera_matrizes() self.assertTrue( (builder.matriz_presencas == MATRIZ_PRESENCAS_ESPERADA).all())
def test_matriz_votacoes(self): # linhas são parlamentares e colunas são votações MATRIZ_VOTACAO_ESPERADA = numpy.matrix( [[1., -1., -1., 1., 1., 1., 1., -1., 1], [0., -1., -1., 1., 1., 1., 1., -1., 1], [-1., -1., 1., 1., 0., 1., 0., -1., 0], [1., -1., -1., 1., 1., 1., 1., 0., 1], [1., -1., -1., 0., 1., 1., 1., -1., 1], [1., -1., -1., -1., 1., 1., 1., -1., 1], [-1., 1., 1., 1., -1., 0., 1., 1., 1], [-1., 1., 1., -1., -1., 1., 0., 0., 0], [-1., 1., 1., 0., -1., 1., 1., 1., 1]]) builder = analise.MatrizesDeDadosBuilder(self.votacoes, self.partidos, self.parlamentares) builder.gera_matrizes() self.assertTrue( (builder.matriz_votacoes == MATRIZ_VOTACAO_ESPERADA).all())
def test_matriz_votacao(self): vetor_girondinos = [ mean([1, 0, -1]), mean([-1, -1, -1]), mean([-1, -1, 1]), mean([1, 1, 1]), mean([1, 1, 0]), mean([1, 1, 1]), mean([1, 1, 0]), mean([-1, -1, -1]) ] vetor_jacobinos = [ mean([1, 1, 1]), mean([-1, -1, -1]), mean([-1, -1, -1]), mean([1, 0 - 1]), mean([1, 1, 1]), mean([1, 1, 1]), mean([1, 1, 1]), mean([0, -1, -1]) ] vetor_monarquistas = [ mean([-1, -1, -1]), mean([1, 1, 1]), mean([1, 1, 1]), mean([1, -1]), mean([-1, -1, -1]), mean([1, 1]), mean([1, 1]), mean([1, 1]) ] MATRIZ_VOTACAO_ESPERADA = numpy.matrix( [vetor_girondinos, vetor_jacobinos, vetor_monarquistas]) builder = analise.MatrizesDeDadosBuilder(self.votacoes, self.partidos, self.legislaturas) builder.gera_matrizes() matriz_votacao_por_partido = builder.matriz_votacoes_por_partido self.assertTrue( (matriz_votacao_por_partido == MATRIZ_VOTACAO_ESPERADA).all())