forked from TadaSoftware/PyNFe
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest_utils.py
242 lines (168 loc) · 8.74 KB
/
test_utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
#!/usr/bin/env python
# *-* encoding: utf8 *-*
import unittest
from pynfe. utils import (
so_numeros,
obter_pais_por_codigo,
normalizar_municipio,
obter_codigo_por_municipio,
obter_municipio_por_codigo,
obter_uf_por_codigo,
formatar_decimal,
remover_acentos
)
from lxml import etree
from pynfe.utils.descompactar import DescompactaGzip
class UtilsTestCase(unittest.TestCase):
# so_numeros
def test_so_numeros_none(self):
self.assertEqual(so_numeros(None), "")
def test_so_numeros_valor_int(self):
self.assertEqual(so_numeros(223), "223")
def test_so_numeros_valor_string(self):
self.assertEqual(so_numeros("223"), "223")
def test_so_numeros_com_letras_e_numeros(self):
self.assertEqual(so_numeros("aa223bb"), "223")
def test_so_numeros_somente_com_letras(self):
self.assertEqual(so_numeros("aabbccdd"), "")
# obter_pais_por_codigo
def test_obter_pais_por_codigo_brasil(self):
self.assertEqual(obter_pais_por_codigo("1058"), "Brasil")
def test_obter_pais_por_codigo_paisesbaixos(self):
self.assertEqual(obter_pais_por_codigo("5738"), None)
# normalizar_municipio
def test_normalizar_municipio_acento_til_na_letra_a(self):
self.assertEqual(normalizar_municipio("São Paulo"), 'SAO PAULO')
def test_normalizar_municipio_cedilha_e_acento_til_na_letra_o(self):
self.assertEqual(normalizar_municipio("Monções"), 'MONCOES')
def test_normalizar_municipio_acento_agudo_na_letra_a(self):
self.assertEqual(normalizar_municipio("Cuiabá"), 'CUIABA')
def test_normalizar_municipio_acento_agudo_na_letra_e(self):
self.assertEqual(normalizar_municipio("Avaré"), 'AVARE')
def test_normalizar_municipio_acento_agudo_na_letra_i(self):
self.assertEqual(normalizar_municipio("Aguaí"), 'AGUAI')
def test_normalizar_municipio_acento_agudo_na_letra_o(self):
self.assertEqual(normalizar_municipio("Altinópolis"), 'ALTINOPOLIS')
def test_normalizar_municipio_acento_agudo_na_letra_u(self):
self.assertEqual(normalizar_municipio("Cabreúva"), 'CABREUVA')
def test_normalizar_municipio_acento_circunflexo_na_letra_a(self):
self.assertEqual(normalizar_municipio("Alvinlândia"), 'ALVINLANDIA')
def test_normalizar_municipio_acento_circunflexo_na_letra_e(self):
self.assertEqual(normalizar_municipio("Governador Eugênio Barros"), 'GOVERNADOR EUGENIO BARROS')
def test_normalizar_municipio_acento_circunflexo_na_letra_o(self):
self.assertEqual(normalizar_municipio("Santo Antônio de Posse"), 'SANTO ANTONIO DE POSSE')
def test_normalizar_municipio_apostrofo(self):
self.assertEqual(normalizar_municipio("Lambari D'Oeste"), 'LAMBARI DOESTE')
# obter_codigo_por_municipio
def test_obter_codigo_por_municipio_saopaulo_3550308(self):
self.assertEqual(obter_codigo_por_municipio("São Paulo", "SP"), '3550308')
def test_fail_obter_codigo_por_municipio_saopaulo_3550308(self):
self.assertNotEqual(obter_codigo_por_municipio("São Paulo", "SP"), '0000000')
def test_obter_codigo_por_municipio_curitiba_4106902(self):
self.assertEqual(obter_codigo_por_municipio("Curitiba", "PR"), '4106902')
def test_fail_obter_codigo_por_municipio_curitiba_4106902(self):
self.assertNotEqual(obter_codigo_por_municipio("Curitiba", "PR"), '0000000')
# obter_municipio_por_codigo
def test_obter_municipio_por_codigo_saopaulo_3550308(self):
self.assertEqual(obter_municipio_por_codigo("3550308", "SP"), 'São Paulo')
def test_obter_municipio_por_codigo_saopaulo_3550308_normalizado(self):
self.assertEqual(obter_municipio_por_codigo("3550308", "SP", True), 'SAO PAULO')
def test_obter_municipio_por_codigo_curitiba_4106902(self):
self.assertEqual(obter_municipio_por_codigo("4106902", "PR"), 'Curitiba')
def test_obter_municipio_por_codigo_curitiba_4106902_normalizado(self):
self.assertEqual(obter_municipio_por_codigo("4106902", "PR", True), 'CURITIBA')
def test_fail_obter_municipio_por_codigo_saopaulo_3550308(self):
with self.assertRaises(Exception):
obter_municipio_por_codigo("0000000", "SP")
self.assertRaises(ValueError)
# obter_uf_por_codigo
def test_obter_uf_por_codigo_ro_11(self):
self.assertEqual(obter_uf_por_codigo("11"), 'RO')
def test_obter_uf_por_codigo_ac_12(self):
self.assertEqual(obter_uf_por_codigo("12"), 'AC')
def test_obter_uf_por_codigo_am_13(self):
self.assertEqual(obter_uf_por_codigo("13"), 'AM')
def test_obter_uf_por_codigo_rr_14(self):
self.assertEqual(obter_uf_por_codigo("14"), 'RR')
def test_obter_uf_por_codigo_pa_15(self):
self.assertEqual(obter_uf_por_codigo("15"), 'PA')
def test_obter_uf_por_codigo_ap_16(self):
self.assertEqual(obter_uf_por_codigo("16"), 'AP')
def test_obter_uf_por_codigo_to_17(self):
self.assertEqual(obter_uf_por_codigo("17"), 'TO')
def test_obter_uf_por_codigo_ma_21(self):
self.assertEqual(obter_uf_por_codigo("21"), 'MA')
def test_obter_uf_por_codigo_pi_22(self):
self.assertEqual(obter_uf_por_codigo("22"), 'PI')
def test_obter_uf_por_codigo_ce_23(self):
self.assertEqual(obter_uf_por_codigo("23"), 'CE')
def test_obter_uf_por_codigo_rn_24(self):
self.assertEqual(obter_uf_por_codigo("24"), 'RN')
def test_obter_uf_por_codigo_pb_25(self):
self.assertEqual(obter_uf_por_codigo("25"), 'PB')
def test_obter_uf_por_codigo_pe_26(self):
self.assertEqual(obter_uf_por_codigo("26"), 'PE')
def test_obter_uf_por_codigo_al_27(self):
self.assertEqual(obter_uf_por_codigo("27"), 'AL')
def test_obter_uf_por_codigo_se_28(self):
self.assertEqual(obter_uf_por_codigo("28"), 'SE')
def test_obter_uf_por_codigo_ba_29(self):
self.assertEqual(obter_uf_por_codigo("29"), 'BA')
def test_obter_uf_por_codigo_mg_31(self):
self.assertEqual(obter_uf_por_codigo("31"), 'MG')
def test_obter_uf_por_codigo_es_32(self):
self.assertEqual(obter_uf_por_codigo("32"), 'ES')
def test_obter_uf_por_codigo_rj_33(self):
self.assertEqual(obter_uf_por_codigo("33"), 'RJ')
def test_obter_uf_por_codigo_pr_41(self):
self.assertEqual(obter_uf_por_codigo("41"), 'PR')
def test_obter_uf_por_codigo_sc_42(self):
self.assertEqual(obter_uf_por_codigo("42"), 'SC')
def test_obter_uf_por_codigo_rs_43(self):
self.assertEqual(obter_uf_por_codigo("43"), 'RS')
def test_obter_uf_por_codigo_ms_50(self):
self.assertEqual(obter_uf_por_codigo("50"), 'MS')
def test_obter_uf_por_codigo_mt_51(self):
self.assertEqual(obter_uf_por_codigo("51"), 'MT')
def test_obter_uf_por_codigo_go_52(self):
self.assertEqual(obter_uf_por_codigo("52"), 'GO')
def test_obter_uf_por_codigo_df_53(self):
self.assertEqual(obter_uf_por_codigo("53"), 'DF')
def test_obter_uf_por_codigo_an_91(self):
self.assertEqual(obter_uf_por_codigo("91"), 'AN')
# formatar_decimal
def test_formatar_decimal_1(self):
self.assertEqual(formatar_decimal(1), '1.00')
def test_formatar_decimal_1_0(self):
self.assertEqual(formatar_decimal(1.0), '1.00')
def test_formatar_decimal_1_00(self):
self.assertEqual(formatar_decimal(1.00), '1.00')
def test_formatar_decimal_1_01(self):
self.assertEqual(formatar_decimal(1.01), '1.01')
def test_formatar_decimal_1_011(self):
self.assertEqual(formatar_decimal(1.011), '1.011')
# remover_acentos
def test_remover_acentos_com_acento(self):
self.assertEqual(remover_acentos("á"), "a")
self.assertEqual(remover_acentos("é"), "e")
self.assertEqual(remover_acentos("í"), "i")
self.assertEqual(remover_acentos("ó"), "o")
self.assertEqual(remover_acentos("ú"), "u")
self.assertEqual(remover_acentos("â"), "a")
self.assertEqual(remover_acentos("ê"), "e")
self.assertEqual(remover_acentos("ô"), "o")
self.assertEqual(remover_acentos("à"), "a")
self.assertEqual(remover_acentos("ã"), "a")
self.assertEqual(remover_acentos("õ"), "o")
self.assertEqual(remover_acentos("a"), "a")
self.assertEqual(remover_acentos("b"), "b")
self.assertEqual(remover_acentos("c"), "c")
class DescompactaGzipUtils(unittest.TestCase):
def test_descompacta_string_gzip(self):
gzip_str = 'H4sIAAAAAAAACrPJS0u1K0ktLklVKKgEsm30QQIABhWzIhYAAAA='
esperado = '<nfe>teste pynfe</nfe>'
descompacta_gzip = DescompactaGzip.descompacta(gzip_str)
resultado_str = etree.tostring(descompacta_gzip).decode('utf-8')
self.assertEqual(resultado_str, esperado)
if __name__ == '__main__':
unittest.main()