🐛 fix for locale constructor (#67)

Author: ianardee

mindee/fields/locale.py

@@ -32,10 +32,7 @@ def __init__(
 
     @staticmethod
     def _get_value(locale_prediction, key: str):
-        if (
-            key not in locale_prediction.keys()
-            or locale_prediction["language"] == "N/A"
-        ):
+        if key not in locale_prediction or locale_prediction[key] == "N/A":
             return None
         return locale_prediction[key]
 

tests/fields/test_locale.py

@@ -23,3 +23,18 @@ def test_constructor_almost_empty_field():
     assert locale.language is None
     assert locale.country is None
     assert locale.currency is None
+
+
+def test_constructor_empty_language():
+    field_dict = {
+        "value": "en-EN",
+        "country": "uk",
+        "currency": "GBP",
+        "language": "N/A",
+        "confidence": 0.1,
+    }
+    locale = Locale(field_dict)
+    assert locale.language is None
+    assert locale.country == "uk"
+    assert locale.currency == "GBP"
+    assert str(locale) == "en-EN; uk; GBP;"