def test_json(self):
EXPECTED_JSON = {
u'periodos': {
'1': {
u'quantidade_votacoes': 8,
u'nome': u'1989 e 1990'
}
},
u'partidos': [{
u'cor': u'#008000',
u'nome': u'Girondinos',
u'tamanho': [[1, 26.0]],
u'numero': 27,
u'y': [[1, 87.62]],
u'x': [[1, 34.15]]
}, {
u'cor': u'#800080',
u'nome': u'Monarquistas',
u'tamanho': [[1, 26.0]],
u'numero': 79,
u'y': [[1, 44.91]],
u'x': [[1, 90.51]]
}, {
u'cor': u'#FF0000',
u'nome': u'Jacobinos',
u'tamanho': [[1, 26.0]],
u'numero': 42,
u'y': [[1, 17.47]],
u'x': [[1, 25.34]]
}]
}
gen = grafico.JsonAnaliseGenerator()
json = gen.get_json_dic(self.casa_legislativa)
self.maxDiff = None
self.assertEqual(json, EXPECTED_JSON)
def test_json(self):
gen = grafico.JsonAnaliseGenerator(self.analiseTemporal)
generated_json = gen.get_json()
dict_analise = json.loads(generated_json)
self.assertEquals(dict_analise['geral']['total_votacoes'], 8)
dict_casa = dict_analise['geral']['CasaLegislativa']
self.assertEquals(dict_casa['nome_curto'], self.casa.nome_curto)
list_periodos = dict_analise['periodos']
self.assertEquals(len(list_periodos), 1)
dict_periodo = list_periodos[0]
self.assertTrue('1989' in dict_periodo['nome'])
list_partidos = dict_analise['partidos']
self.assertEquals(len(list_partidos), 3)
dict_partido = list_partidos[0]
toutes_les_parties = [
conv.JACOBINOS, conv.GIRONDINOS, conv.MONARQUISTAS
]
self.assertTrue(dict_partido['nome'] in toutes_les_parties)
list_tamanhos = dict_partido['t']
self.assertEquals(list_tamanhos[0], 3)
list_parlamentares = dict_partido['parlamentares']
self.assertEquals(len(list_parlamentares), 3)
dict_parlamentar = list_parlamentares[0]
list_xs = dict_parlamentar['x']
x = list_xs[0]
self.assertTrue(x >= 0 and x <= 100)
def setUp(self):
self.casa = models.CasaLegislativa.objects.get(nome_curto='conv')
for partido in JsonAnaliseGeneratorTest.importer.partidos:
if partido.nome == conv.GIRONDINOS:
self.girondinos = partido
if partido.nome == conv.JACOBINOS:
self.jacobinos = partido
if partido.nome == conv.MONARQUISTAS:
self.monarquistas = partido
analise_temporal = AnaliseTemporal()
analise_temporal.casa_legislativa = self.casa
analise_temporal.periodicidade = models.BIENIO
analise_temporal.analises_periodo = []
analise_temporal.total_votacoes = 8
self.chefe = models.ChefeExecutivo(nome="Luiz Inacio Pierre da Silva", genero="M", partido = self.girondinos,
mandato_ano_inicio = 1989, mandato_ano_fim = 1990)
self.chefe.save()
self.chefe.casas_legislativas.add(self.casa)
ap1 = AnalisePeriodo()
periodos_retriever = utils.PeriodosRetriever(self.casa, models.BIENIO)
periodos = periodos_retriever.get_periodos()
ap1.casa_legislativa = None
ap1.periodo = periodos[0]
ap1.partidos = [self.girondinos, self.jacobinos, self.monarquistas]
ap1.votacoes = []
ap1.chefes_executivos = [self.chefe]
ap1.num_votacoes = 0
ap1.tamanhos_partidos = {
self.girondinos: 3, self.jacobinos: 3, self.monarquistas: 3}
ap1.soma_dos_tamanhos_dos_partidos = 3 * 3
ap1.pca = PCAStub()
ap1.coordenadas_partidos = {}
ap1.coordenadas_partidos[conv.JACOBINOS] = [-0.4, 0.3]
ap1.coordenadas_partidos[conv.GIRONDINOS] = [0.9, -0.3]
ap1.coordenadas_partidos[conv.MONARQUISTAS] = [0.2, 0.1]
ap1.parlamentares_por_partido = \
JsonAnaliseGeneratorTest.importer.parlamentares
ap1.coordenadas_parlamentares = {} # parlamentar.id => [x,y]
for partido, parlamentares in ap1.parlamentares_por_partido.items():
for parlamentar in parlamentares:
ap1.coordenadas_parlamentares[parlamentar.id] = [random(),
random()]
analise_temporal.analises_periodo.append(ap1)
gen = grafico.JsonAnaliseGenerator(analise_temporal)
generated_json = gen.get_json()
self.dict_analise = json.loads(generated_json)
def test_json(self):
EXPECTED_JSON = {
u'periodos': {
'1': {
u'quantidade_votacoes': 4,
u'nome': u'1989 1o Semestre'
},
'2': {
u'quantidade_votacoes': 4,
u'nome': u'1989 2o Semestre'
}
},
u'partidos': [{
u'cor': u'#000000',
u'nome': u'Girondinos',
u'tamanho': [[1, 40.4], [2, 40.4]],
u'numero': 27,
u'y': [[1, 10.94], [2, 14.8]],
u'x': [[1, 38.12], [2, 29.33]]
}, {
u'cor': u'#000000',
u'nome': u'Monarquistas',
u'tamanho': [[1, 40.4], [2, 40.4]],
u'numero': 79,
u'y': [[1, 59.24], [2, 49.7]],
u'x': [[1, 89.77], [2, 90.82]]
}, {
u'cor': u'#000000',
u'nome': u'Jacobinos',
u'tamanho': [[1, 40.4], [2, 40.4]],
u'numero': 42,
u'y': [[1, 79.82], [2, 85.51]],
u'x': [[1, 22.12], [2, 29.85]]
}]
}
gen = grafico.JsonAnaliseGenerator()
json = gen.get_json_dic(self.casa_legislativa)
self.maxDiff = None
self.assertEqual(json, EXPECTED_JSON)