refactor: módulo uf.py

Author: camilamaia

brutils/data/enums/__init__.py

@@ -1 +1 @@
-from .uf import CODE_TO_UF, UF
+from brutils.data.enums.uf import UF, UF_CODE

brutils/data/enums/uf.py

@@ -1,4 +1,4 @@
-from .better_enum import BetterEnum
+from brutils.data.enums.better_enum import BetterEnum
 
 
 class UF(BetterEnum):
@@ -31,7 +31,7 @@ class UF(BetterEnum):
     TO = "Tocantins"
 
 
-class CODE_TO_UF(BetterEnum):
+class UF_CODE(BetterEnum):
     AC = "12"
     AL = "27"
     AP = "16"
@@ -43,7 +43,7 @@ class CODE_TO_UF(BetterEnum):
     GO = "52"
     MA = "21"
     MT = "51"
-    MS = "52"
+    MS = "50"
     MG = "31"
     PA = "15"
     PB = "25"

brutils/ibge/uf.py

@@ -1,7 +1,7 @@
-from brutils.data.enums.uf import CODE_TO_UF, UF
+from brutils.data.enums.uf import UF, UF_CODE
 
 
-def convert_code_to_uf(code):  # type: (str) -> str | None
+def convert_code_to_uf(code: str) -> str | None:
     """
     Converts a given IBGE code (2-digit string) to its corresponding UF (state abbreviation).
 
@@ -23,13 +23,10 @@ def convert_code_to_uf(code):  # type: (str) -> str | None
         >>> convert_code_to_uf('99')
         >>>
     """
+    if code not in UF_CODE.values:
+        return None
 
-    result = None
-
-    if code in CODE_TO_UF.values:
-        result = CODE_TO_UF(code).name
-
-    return result
+    return UF_CODE(code).name
 
 
 def convert_uf_to_name(uf: str) -> str | None:
@@ -51,12 +48,12 @@ def convert_uf_to_name(uf: str) -> str | None:
         >>> convert_uf_to_name('rj')
         'Rio de Janeiro'
     """
-    if not uf or not isinstance(uf, str):
+    if not uf or not isinstance(uf, str) or len(uf.strip()) != 2:
         return None
 
     federal_unit = uf.strip().upper()
 
-    if federal_unit not in UF.__members__:
+    if federal_unit not in UF.names:
         return None
 
     result = UF[federal_unit].value

tests/ibge/test_uf.py

@@ -27,4 +27,5 @@ def test_convert_uf_to_name(self):
 
         # Testes para códigos inválidos
         self.assertIsNone(convert_code_to_uf("XX"))
+        self.assertIsNone(convert_code_to_uf("XXX"))
         self.assertIsNone(convert_code_to_uf(""))