diff --git a/mindee/fields/locale.py b/mindee/fields/locale.py index 76de912c..49347a8d 100644 --- a/mindee/fields/locale.py +++ b/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] diff --git a/tests/fields/test_locale.py b/tests/fields/test_locale.py index cb5ff205..5493fadb 100644 --- a/tests/fields/test_locale.py +++ b/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;"