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
Exemplo n.º 2
0
    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