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())