feat: add remove_bank function to manage bank entries

Author: gabino

clabe/__init__.py

@@ -8,6 +8,7 @@
     'get_bank_name',
     'validate_clabe',
     'configure_additional_bank',
+    'remove_bank',
 ]
 
 from .banks import BANK_NAMES, BANKS
@@ -17,6 +18,7 @@
     configure_additional_bank,
     generate_new_clabes,
     get_bank_name,
+    remove_bank,
     validate_clabe,
 )
 from .version import __version__

clabe/validations.py

@@ -98,3 +98,18 @@ def configure_additional_bank(bank_code_banxico: str, bank_name: str) -> None:
     )
     BANKS[request.bank_code_abm] = request.bank_code_banxico
     BANK_NAMES[request.bank_code_banxico] = request.bank_name
+
+
+def remove_bank(bank_code_banxico: str) -> None:
+    bank_code_abm = next(
+        (
+            abm
+            for abm, banxico in BANKS.items()
+            if banxico == bank_code_banxico
+        ),
+        None,
+    )
+
+    if bank_code_abm:
+        BANKS.pop(bank_code_abm)
+        BANK_NAMES.pop(bank_code_banxico)

tests/test_clabe.py

@@ -5,6 +5,7 @@
     configure_additional_bank,
     generate_new_clabes,
     get_bank_name,
+    remove_bank,
     validate_clabe,
 )
 
@@ -64,3 +65,9 @@ def test_configure_additional_bank_success(abm_code, banxico_code, name):
 def test_configure_additional_bank_invalid_inputs(banxico_code, name):
     with pytest.raises(ValueError):
         configure_additional_bank(banxico_code, name)
+
+
+def test_remove_bank():
+    remove_bank('40138')
+    with pytest.raises(ValueError):
+        get_bank_name('138')