def test_get_empenhado_gnd_list_when_total_is_none(self):
gnds_dicts = [
{'gnd_geologia__desc': 'gnd1', 'gnd_geologia__slug': 'g1',
'empenhado': None},
{'gnd_geologia__desc': 'gnd2', 'gnd_geologia__slug': 'g2',
'empenhado': None},
]
empenhado_total = None
serializer = GeologiaSerializer([])
assert serializer._get_empenhado_gnds_list(gnds_dicts, empenhado_total)
def test_get_empenhado_gnds_list(self, empenhado_fixture):
gnds, empenhado_total = empenhado_fixture
expected = []
for gnd in gnds:
if gnd['empenhado'] is None:
gnd['empenhado'] = 0
expected.append({
"name": gnd['gnd_geologia__desc'],
"slug": gnd['gnd_geologia__slug'],
"value": gnd['empenhado'],
"percent": gnd['empenhado'] / empenhado_total
})
serializer = GeologiaSerializer([])
ret = serializer._get_empenhado_gnds_list(
gnds, empenhado_total)
assert expected == ret
def test_get_empenhado_gnds_list(self, empenhado_fixture):
gnds, empenhado_total = empenhado_fixture
deflator = Deflator.objects.get(year__year=2018)
expected = []
for gnd in gnds:
if gnd['empenhado'] is None:
gnd['empenhado'] = 0
empenhado = gnd['empenhado'] / deflator.index_number
expected.append({
"name": gnd['gnd_geologia__desc'],
"slug": gnd['gnd_geologia__slug'],
"value": empenhado,
"percent": empenhado / empenhado_total
})
serializer = GeologiaSerializer([])
ret = serializer._get_empenhado_gnds_list(gnds, empenhado_total,
deflator.year)
assert expected == ret