2569718930@qq.com
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
from src.data_collection.russia_station_sources import (
|
|
RUSSIA_MOSCOW_MAP_STATION_LIMIT,
|
|
RUSSIA_MOSCOW_STATIONS,
|
|
RussiaStationSourceMixin,
|
|
)
|
|
|
|
|
|
def test_moscow_station_registry_includes_vnukovo_nearby_ring():
|
|
assert "27524" in RUSSIA_MOSCOW_STATIONS
|
|
assert "27500" in RUSSIA_MOSCOW_STATIONS
|
|
assert RUSSIA_MOSCOW_STATIONS["27500"]["station_label"] == "Tolstopaltsevo"
|
|
assert RUSSIA_MOSCOW_MAP_STATION_LIMIT == 10
|
|
|
|
|
|
def test_fetch_moscow_official_nearby_keeps_nearest_ten_station_rows():
|
|
class FakeRussiaSource(RussiaStationSourceMixin):
|
|
CITY_REGISTRY = {
|
|
"moscow": {
|
|
"lat": 55.5915,
|
|
"lon": 37.2615,
|
|
}
|
|
}
|
|
|
|
def _ru_cached_station_current(self, station_code, station_meta, use_fahrenheit=False):
|
|
return {
|
|
"station_code": station_code,
|
|
"station_label": station_meta["station_label"],
|
|
"name": station_meta["station_label"],
|
|
"lat": station_meta["lat"],
|
|
"lon": station_meta["lon"],
|
|
"temp": 10.0,
|
|
"source_code": "ru_station_web",
|
|
}
|
|
|
|
rows = FakeRussiaSource().fetch_russia_moscow_official_nearby("moscow")
|
|
|
|
assert len(rows) == RUSSIA_MOSCOW_MAP_STATION_LIMIT
|
|
assert rows[0]["station_code"] == "27524"
|
|
assert rows[1]["station_code"] == "27500"
|
|
assert all(row["distance_km"] is not None for row in rows)
|
|
|
|
|
|
def test_russia_weather_archive_parser_uses_latest_valid_row():
|
|
html = """
|
|
<table>
|
|
<tr><th>hour</th><th>date</th></tr>
|
|
<tr><td>09</td><td>11.04</td></tr>
|
|
<tr><td>21</td><td>17.04</td></tr>
|
|
</table>
|
|
<table>
|
|
<tr><th>a</th><th>b</th><th>c</th><th>d</th><th>cloud</th><th>temp</th></tr>
|
|
<tr><td>x</td><td>x</td><td>x</td><td>x</td><td>x</td><td>+4.2</td></tr>
|
|
<tr><td>x</td><td>x</td><td>x</td><td>x</td><td>x</td><td>+5.8</td></tr>
|
|
</table>
|
|
"""
|
|
|
|
parsed = RussiaStationSourceMixin()._ru_parse_station_current_from_weather_html(html)
|
|
|
|
assert parsed["temp_c"] == 5.8
|
|
assert parsed["raw_hour"] == "21"
|
|
assert parsed["raw_day_month"] == "17.04"
|