Merge pull request #42 from alvarofpp/refactor-renavam

Refactor renavam
alvarofpp · Oct 26, 2021 · d3e03f4 · d3e03f4
2 parents 37f0c2d + ebf9459
commit d3e03f4
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -15,12 +15,13 @@ A documentação pode ser acessada [clicando aqui](https://alvarofpp.github.io/v
 ## Documentos
 Documentos que estão no pacote:
 
-- [CPF](https://github.com/alvarofpp/validate-docbr/wiki/CPF): Cadastro de Pessoas Físicas;
-- [CNH](#): Carteira Nacional de Habilitação;
-- [CNPJ](https://github.com/alvarofpp/validate-docbr/wiki/CNPJ): Cadastro Nacional da Pessoa Jurídica;
-- [CNS](https://github.com/alvarofpp/validate-docbr/wiki/CNS): Cartão Nacional de Saúde;
-- [PIS](#): PIS/NIS/PASEP/NIT;
-- [Título eleitoral](#): Cadastro que permite cidadãos brasileiros votar.
+- [CPF](validate_docbr/CPF.py): Cadastro de Pessoas Físicas;
+- [CNH](validate_docbr/CNH.py): Carteira Nacional de Habilitação;
+- [CNPJ](validate_docbr/CNPJ.py): Cadastro Nacional da Pessoa Jurídica;
+- [CNS](validate_docbr/CNS.py): Cartão Nacional de Saúde;
+- [PIS](validate_docbr/PIS.py): PIS/NIS/PASEP/NIT;
+- [Título eleitoral](validate_docbr/TituloEleitoral.py): Cadastro que permite cidadãos brasileiros votar;
+- [RENAVAM](validate_docbr/RENAVAM.py): Registro Nacional de Veículos Automotores.
 
 Para entender melhor os documentos e suas respectivas classes, basta acessar a [Wiki do projeto](https://github.com/alvarofpp/validate-docbr/wiki).
 

diff --git a/tests/test_all.py b/tests/test_all.py
@@ -6,6 +6,7 @@
 import tests.test_cns
 import tests.test_cnpj
 import tests.test_pis
+import tests.test_renavam
 import tests.test_titulo_eleitor
 import tests.test_generic
 
@@ -20,6 +21,7 @@ def suite():
     test_suite.addTests(loader.loadTestsFromModule(tests.test_cns))
     test_suite.addTests(loader.loadTestsFromModule(tests.test_cnpj))
     test_suite.addTests(loader.loadTestsFromModule(tests.test_pis))
+    test_suite.addTests(loader.loadTestsFromModule(tests.test_renavam))
     test_suite.addTests(loader.loadTestsFromModule(tests.test_titulo_eleitor))
     test_suite.addTests(loader.loadTestsFromModule(tests.test_generic))
 

diff --git a/tests/test_renavam.py b/tests/test_renavam.py
@@ -13,9 +13,9 @@ def test_generate_validate(self):
         """Verifica os métodos de geração e validação de documento."""
         # generate_list
         renavams = (
-            self.renavam.generate_list(1)
-            + self.renavam.generate_list(1, mask=True)
-            + self.renavam.generate_list(1, mask=True, repeat=True)
+                self.renavam.generate_list(1)
+                + self.renavam.generate_list(1, mask=True)
+                + self.renavam.generate_list(1, mask=True, repeat=True)
         )
         self.assertIsInstance(renavams, list)
         self.assertTrue(len(renavams) == 3)

diff --git a/validate_docbr/RENAVAM.py b/validate_docbr/RENAVAM.py
@@ -33,7 +33,7 @@ def generate(self, mask: bool = False) -> str:
 
     def mask(self, doc: str = '') -> str:
         """Coloca a máscara de Renavam na variável doc."""
-        return f"{doc[:10]}-{doc[10]}"
+        return "{}-{}".format(doc[:10], doc[10])
 
     def _generate_last_digit(self, doc: Union[str, list]) -> str:
         """Gerar o dígito verificador do Renavam."""