PolyWeather/tests/test_analysis_ttl.py

from web.analysis_service import _analysis_ttl_for_city, _mgm_hourly_high
from web.core import CACHE_TTL


def test_high_frequency_airport_cities_use_one_minute_analysis_cache():
    assert _analysis_ttl_for_city("chongqing") == 60
    assert _analysis_ttl_for_city("shanghai") == 60
    assert _analysis_ttl_for_city("singapore") == 60


def test_non_high_frequency_city_keeps_default_analysis_cache_ttl():
    assert _analysis_ttl_for_city("new york") == CACHE_TTL


def test_mgm_hourly_high_fills_turkish_model_support():
    assert _mgm_hourly_high(
        {
            "hourly": [
                {"time": "2026-05-17T10:00:00Z", "temp": 18.0},
                {"time": "2026-05-17T11:00:00Z", "temp": 21.5},
                {"time": "2026-05-17T12:00:00Z", "temp": None},
            ]
        }
    ) == 21.5