ajustar nome funcao #220

brazilian-utils · Feb 9, 2024 · 7f252c9 · 7f252c9
1 parent 5ec9d99
commit 7f252c9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/brutils/voter_id.py b/brutils/voter_id.py
@@ -214,7 +214,7 @@ def _calculate_vd2(federative_union, vd1):  # type: (str, int) -> str
     return vd2
 
 
-def generate_voter_id(state="ZZ") -> str:
+def generate(state="ZZ") -> str:
     """
     Generates a random valid Brazilian voter registration.
 

diff --git a/tests/test_voter_id.py b/tests/test_voter_id.py
@@ -7,7 +7,7 @@
     _get_sequential_number,
     _get_verifying_digits,
     _is_length_valid,
-    generate_voter_id,
+    generate,
     is_valid,
 )
 
@@ -90,22 +90,22 @@ def test_calculate_vd2(self):
 
     def test_generate_voter_id(self):
         # test if is_valid a voter id from MG
-        voter_id = generate_voter_id(state="MG")
+        voter_id = generate(state="MG")
 
         self.assertIs(is_valid(voter_id), True)
 
         # test if is_valid a voter id from AC
-        voter_id = generate_voter_id(state="AC")
+        voter_id = generate(state="AC")
 
         self.assertIs(is_valid(voter_id), True)
 
         # test if is_valid a voter id from foreigner
-        voter_id = generate_voter_id()
+        voter_id = generate()
 
         self.assertIs(is_valid(voter_id), True)
 
         # test if UF is not valid
-        voter_id = generate_voter_id(state="XX")
+        voter_id = generate(state="XX")
 
         self.assertIs(is_valid(voter_id), False)