def test_get_orcado_gnds_list(self, orcado_fixture): gnds, orcado_total = orcado_fixture expected = [ { "name": gnd['gnd_geologia__desc'], "slug": gnd['gnd_geologia__slug'], "value": gnd['orcado'], "percent": gnd['orcado'] / orcado_total } for gnd in gnds ] serializer = GeologiaSerializer([]) ret = serializer._get_orcado_gnds_list(gnds, orcado_total) assert expected == ret
def test_get_orcado_gnds_list(self, orcado_fixture): gnds, orcado_total = orcado_fixture deflator = Deflator.objects.get(year__year=2017) expected = [] for gnd in gnds: orcado = gnd['orcado'] / deflator.index_number gnd_dict = { "name": gnd['gnd_geologia__desc'], "slug": gnd['gnd_geologia__slug'], "value": orcado, "percent": orcado / orcado_total } expected.append(gnd_dict) serializer = GeologiaSerializer([]) ret = serializer._get_orcado_gnds_list(gnds, orcado_total, deflator.year) assert expected == ret