fixed issue converting regimen

SAT-CFDI · Oct 12, 2023 · 86e7778 · 86e7778
1 parent 6260f54
commit 86e7778
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/satcfdi/csf/__init__.py b/satcfdi/csf/__init__.py
@@ -36,6 +36,7 @@ def _request_constancia(rfc: str, id_cif: str):
 
 
 def _find_regimen(regimen):
+    regimen = regimen.rstrip('.')
     for k, v in select_all('C756_c_RegimenFiscal').items():
         if regimen.endswith(v):
             return Code(k, v)

diff --git a/tests/test_csf.py b/tests/test_csf.py
@@ -6,7 +6,7 @@
 import pytest
 
 
-from satcfdi.csf import retrieve, _parse_response
+from satcfdi.csf import retrieve, _parse_response, _find_regimen
 from tests.utils import verify_result
 
 current_dir = os.path.dirname(__file__)
@@ -66,3 +66,8 @@ def test_parse_response(file):
     res = _parse_response(data)
     pprint(res)
 
+
+def test_regimen_name():
+    res = _find_regimen('Régimen de las Actividades Empresariales con ingresos a través de Plataformas Tecnológicas.')
+    assert res == '625'
+