Merge pull request #291 from leogregianin/main

Adiciona cidades

Author: juniortada

pynfe/data/MunIBGE/MunIBGE-UF15.txt

@@ -4,6 +4,7 @@
 1500305	Afuá
 1500347	Água Azul do Norte
 1500404	Alenquer
+1504752	Mojuí dos Campos
 1500503	Almeirim
 1500602	Altamira
 1500701	Anajás

pynfe/data/MunIBGE/MunIBGE-UF42.txt

@@ -190,6 +190,7 @@
 4212403	Pedras Grandes
 4212502	Penha
 4212601	Peritiba
+4212650	Pescaria Brava
 4212700	Petrolândia
 4212809	Balneário Piçarras
 4212908	Pinhalzinho
@@ -291,3 +292,4 @@
 4219606	Xavantina
 4219705	Xaxim
 4219853	Zortéa
+4220000	Balneário Rincão

pynfe/data/MunIBGE/MunIBGE-UF43.txt

@@ -315,6 +315,7 @@
 4314472	Pinhal Grande
 4314498	Pinheirinho do Vale
 4314506	Pinheiro Machado
+4314548	Pinto Bandeira
 4314555	Pirapó
 4314605	Piratini
 4314704	Planalto

pynfe/data/MunIBGE/MunIBGE-UF50.txt

@@ -55,6 +55,7 @@
 5006002	Nova Alvorada do Sul
 5006200	Nova Andradina
 5006259	Novo Horizonte do Sul
+5006275	Paraíso das Águas
 5006309	Paranaíba
 5006358	Paranhos
 5006408	Pedro Gomes

tests/test_utils.py

@@ -12,6 +12,7 @@
     obter_uf_por_codigo,
     formatar_decimal,
     remover_acentos,
+    carregar_arquivo_municipios
 )
 from lxml import etree
 from pynfe.utils.descompactar import DescompactaGzip
@@ -241,5 +242,112 @@ def test_descompacta_string_gzip(self):
         self.assertEqual(resultado_str, esperado)
 
 
+class CidadesTestCase(unittest.TestCase):
+    """
+    Testar quantidade de cidades por UF
+    pelo arquivos MunIBGE-UFXX.txt
+
+    """
+    def test_quantidade_cidades_ac(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=12)), 22)
+
+    def test_quantidade_cidades_al(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=27)), 102)
+
+    def test_quantidade_cidades_ap(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=16)), 16)
+
+    def test_quantidade_cidades_am(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=13)), 62)
+
+    def test_quantidade_cidades_ba(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=29)), 417)
+
+    def test_quantidade_cidades_ce(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=23)), 184)
+
+    def test_quantidade_cidades_df(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=53)), 1)
+
+    def test_quantidade_cidades_es(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=32)), 78)
+
+    def test_quantidade_cidades_go(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=52)), 246)
+
+    def test_quantidade_cidades_ma(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=21)), 217)
+
+    def test_quantidade_cidades_mt(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=51)), 141)
+
+    def test_quantidade_cidades_ms(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=50)), 79)
+
+    def test_quantidade_cidades_mg(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=31)), 853)
+
+    def test_quantidade_cidades_pa(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=15)), 144)
+
+    def test_quantidade_cidades_pb(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=25)), 223)
+
+    def test_quantidade_cidades_pr(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=41)), 399)
+
+    def test_quantidade_cidades_pe(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=26)), 185)
+
+    def test_quantidade_cidades_pi(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=22)), 224)
+
+    def test_quantidade_cidades_rj(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=33)), 92)
+
+    def test_quantidade_cidades_rn(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=24)), 167)
+
+    def test_quantidade_cidades_rs(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=43)), 497)
+
+    def test_quantidade_cidades_ro(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=11)), 52)
+
+    def test_quantidade_cidades_rr(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=14)), 15)
+
+    def test_quantidade_cidades_sc(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=42)), 295)
+
+    def test_quantidade_cidades_sp(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=35)), 645)
+
+    def test_quantidade_cidades_se(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=28)), 75)
+
+    def test_quantidade_cidades_to(self):
+        self.assertEqual(len(carregar_arquivo_municipios(uf=17)), 139)
+
+    def test_quantidade_total_cidades(self):
+        """
+        Carregar todos os arquivos
+        e verificar a quantidade total de cidades
+
+        """
+        total = 0
+        lista_de_codigos_uf = [
+            11, 12, 13, 14, 15, 16, 17,  # Norte
+            21, 22, 23, 24, 25, 26, 27, 28, 29,  # Nordeste
+            31, 32, 33, 35,  # Sudeste
+            41, 42, 43,  # Sul
+            50, 51, 52, 53  # Centro-Oeste
+        ]
+        for uf in lista_de_codigos_uf:
+            total += len(carregar_arquivo_municipios(uf=uf))
+
+        self.assertEqual(total, 5570)
+
+
 if __name__ == "__main__":
     unittest.main()