diff --git a/tests/test_pogoda.py b/tests/test_pogoda.py index d6486ab..41c916e 100644 --- a/tests/test_pogoda.py +++ b/tests/test_pogoda.py @@ -163,11 +163,11 @@ class TestTranslateWeather: [ ("", "—"), (None, "—"), - (" ", " "), # пробелы не считаются пустыми + (" ", "—"), # строка из пробелов тоже считается пустой ], ) def test_translate_empty(self, input_value, expected) -> None: - """Пустой или None ввод должен возвращать '—'.""" + """Пустой, None или только пробелы должен возвращать '—'.""" assert translate_weather(input_value) == expected def test_translate_unknown_returns_original(self) -> None: diff --git a/utils/pogoda.py b/utils/pogoda.py index 1a5fc93..6734590 100644 --- a/utils/pogoda.py +++ b/utils/pogoda.py @@ -172,9 +172,9 @@ _WEATHER_MAPPING = [ def translate_weather(en: Optional[str]) -> str: - if not en: + if not en or not en.strip(): return "—" - en_lower = en.lower() + en_lower = en.strip().lower() for key, value in _WEATHER_MAPPING: if key.lower() in en_lower: return value