344 lines
11 KiB
Python
344 lines
11 KiB
Python
"""Tests for car journey YAML generation."""
|
|
|
|
import json
|
|
from datetime import date, datetime
|
|
from pathlib import Path
|
|
|
|
import agenda.car_journey_yaml
|
|
from agenda.types import Trip
|
|
|
|
|
|
def write_home_route(data_dir: Path) -> None:
|
|
"""Write a route that identifies PCH as the home coordinate."""
|
|
route_dir = data_dir / "car_routes"
|
|
route_dir.mkdir()
|
|
(route_dir / "PCH_to_EMF.geojson").write_text(
|
|
json.dumps(
|
|
{
|
|
"type": "Feature",
|
|
"properties": {},
|
|
"geometry": {
|
|
"type": "LineString",
|
|
"coordinates": [[-2.60283, 51.44083], [-2.37789, 52.038]],
|
|
},
|
|
}
|
|
)
|
|
)
|
|
|
|
|
|
def test_car_journeys_for_trip_uses_accommodation_dates_and_locations(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Build outbound and return journeys for the trip accommodation."""
|
|
write_home_route(tmp_path)
|
|
trip = Trip(
|
|
start=date(2025, 4, 22),
|
|
accommodation=[
|
|
{
|
|
"location": "Callestick",
|
|
"from": datetime(2025, 4, 22, 15, 0),
|
|
"to": datetime(2025, 4, 25, 10, 0),
|
|
"latitude": 50.307093,
|
|
"longitude": -5.13466,
|
|
}
|
|
],
|
|
)
|
|
|
|
journeys = agenda.car_journey_yaml.car_journeys_for_trip(
|
|
trip, tmp_path, destination="accommodation"
|
|
)
|
|
|
|
assert [journey.route for journey in journeys] == [
|
|
"PCH_to_Callestick",
|
|
"Callestick_to_PCH",
|
|
]
|
|
assert [journey.depart for journey in journeys] == [
|
|
date(2025, 4, 22),
|
|
date(2025, 4, 25),
|
|
]
|
|
assert journeys[0].start == (-2.60283, 51.44083)
|
|
assert journeys[0].end == (-5.13466, 50.307093)
|
|
assert journeys[1].start == (-5.13466, 50.307093)
|
|
assert journeys[1].end == (-2.60283, 51.44083)
|
|
|
|
|
|
def test_car_journeys_for_trip_can_route_to_airport(tmp_path: Path) -> None:
|
|
"""Build outbound and return journeys for airport access."""
|
|
write_home_route(tmp_path)
|
|
trip = Trip(
|
|
start=date(2025, 11, 17),
|
|
travel=[
|
|
{
|
|
"type": "flight",
|
|
"depart": datetime(2025, 11, 17, 17, 55),
|
|
"arrive": datetime(2025, 11, 18, 14, 50),
|
|
"from_airport": {
|
|
"iata": "LHR",
|
|
"latitude": 51.4775,
|
|
"longitude": -0.461389,
|
|
},
|
|
"to_airport": {
|
|
"iata": "HKG",
|
|
"latitude": 22.308889,
|
|
"longitude": 113.914444,
|
|
},
|
|
},
|
|
{
|
|
"type": "flight",
|
|
"depart": datetime(2025, 11, 30, 22, 45),
|
|
"arrive": datetime(2025, 12, 1, 5, 35),
|
|
"from_airport": {
|
|
"iata": "HKG",
|
|
"latitude": 22.308889,
|
|
"longitude": 113.914444,
|
|
},
|
|
"to_airport": {
|
|
"iata": "LHR",
|
|
"latitude": 51.4775,
|
|
"longitude": -0.461389,
|
|
},
|
|
},
|
|
],
|
|
)
|
|
|
|
journeys = agenda.car_journey_yaml.car_journeys_for_trip(
|
|
trip, tmp_path, destination="airport"
|
|
)
|
|
|
|
assert [journey.route for journey in journeys] == ["PCH_to_LHR", "LHR_to_PCH"]
|
|
assert [journey.depart for journey in journeys] == [
|
|
date(2025, 11, 17),
|
|
date(2025, 12, 1),
|
|
]
|
|
assert journeys[0].start == (-2.60283, 51.44083)
|
|
assert journeys[0].end == (-0.461389, 51.4775)
|
|
assert journeys[1].start == (-0.461389, 51.4775)
|
|
assert journeys[1].end == (-2.60283, 51.44083)
|
|
|
|
|
|
def test_car_journeys_for_trip_can_route_around_ferry(tmp_path: Path) -> None:
|
|
"""Build home, ferry terminal and accommodation driving legs."""
|
|
write_home_route(tmp_path)
|
|
trip = Trip(
|
|
start=date(2025, 7, 4),
|
|
travel=[
|
|
{
|
|
"type": "ferry",
|
|
"depart": datetime(2025, 7, 4, 22, 0),
|
|
"arrive": datetime(2025, 7, 5, 8, 0),
|
|
"from": "Plymouth",
|
|
"to": "Roscoff",
|
|
"from_terminal": {
|
|
"name": "Plymouth",
|
|
"latitude": 50.3651254,
|
|
"longitude": -4.1578164,
|
|
},
|
|
"to_terminal": {
|
|
"name": "Roscoff",
|
|
"latitude": 48.721672,
|
|
"longitude": -3.966925,
|
|
},
|
|
},
|
|
{
|
|
"type": "ferry",
|
|
"depart": datetime(2025, 7, 21, 15, 0),
|
|
"arrive": datetime(2025, 7, 21, 20, 10),
|
|
"from": "Roscoff",
|
|
"to": "Plymouth",
|
|
"from_terminal": {
|
|
"name": "Roscoff",
|
|
"latitude": 48.721672,
|
|
"longitude": -3.966925,
|
|
},
|
|
"to_terminal": {
|
|
"name": "Plymouth",
|
|
"latitude": 50.3651254,
|
|
"longitude": -4.1578164,
|
|
},
|
|
},
|
|
],
|
|
accommodation=[
|
|
{
|
|
"location": "Brest",
|
|
"from": datetime(2025, 7, 5, 16, 0),
|
|
"to": datetime(2025, 7, 21, 10, 0),
|
|
"latitude": 48.3645307,
|
|
"longitude": -4.5534685,
|
|
}
|
|
],
|
|
)
|
|
|
|
journeys = agenda.car_journey_yaml.car_journeys_for_trip(
|
|
trip, tmp_path, destination="ferry"
|
|
)
|
|
|
|
assert [journey.route for journey in journeys] == [
|
|
"PCH_to_Plymouth",
|
|
"Roscoff_to_Brest",
|
|
"Brest_to_Roscoff",
|
|
"Plymouth_to_PCH",
|
|
]
|
|
assert [journey.depart for journey in journeys] == [
|
|
date(2025, 7, 4),
|
|
date(2025, 7, 5),
|
|
date(2025, 7, 21),
|
|
date(2025, 7, 21),
|
|
]
|
|
assert journeys[0].start == (-2.60283, 51.44083)
|
|
assert journeys[0].end == (-4.1578164, 50.3651254)
|
|
assert journeys[1].start == (-3.966925, 48.721672)
|
|
assert journeys[1].end == (-4.5534685, 48.3645307)
|
|
assert journeys[2].start == (-4.5534685, 48.3645307)
|
|
assert journeys[2].end == (-3.966925, 48.721672)
|
|
assert journeys[3].start == (-4.1578164, 50.3651254)
|
|
assert journeys[3].end == (-2.60283, 51.44083)
|
|
|
|
|
|
def test_car_journeys_for_trip_can_route_fly_drive_chain(tmp_path: Path) -> None:
|
|
"""Build home, destination rental and return airport driving legs."""
|
|
write_home_route(tmp_path)
|
|
trip = Trip(
|
|
start=date(2025, 9, 13),
|
|
travel=[
|
|
{
|
|
"type": "flight",
|
|
"depart": datetime(2025, 9, 14, 6, 50),
|
|
"arrive": datetime(2025, 9, 14, 8, 55),
|
|
"from_airport": {
|
|
"iata": "LTN",
|
|
"latitude": 51.8747,
|
|
"longitude": -0.3683,
|
|
"country": "gb",
|
|
},
|
|
"to_airport": {
|
|
"iata": "KEF",
|
|
"latitude": 63.985,
|
|
"longitude": -22.605556,
|
|
"country": "is",
|
|
},
|
|
},
|
|
{
|
|
"type": "flight",
|
|
"depart": datetime(2025, 10, 1, 9, 0),
|
|
"arrive": datetime(2025, 10, 1, 13, 0),
|
|
"from_airport": {
|
|
"iata": "KEF",
|
|
"latitude": 63.985,
|
|
"longitude": -22.605556,
|
|
"country": "is",
|
|
},
|
|
"to_airport": {
|
|
"iata": "LTN",
|
|
"latitude": 51.8747,
|
|
"longitude": -0.3683,
|
|
"country": "gb",
|
|
},
|
|
},
|
|
],
|
|
accommodation=[
|
|
{
|
|
"location": "Luton Airport",
|
|
"country": "gb",
|
|
"from": datetime(2025, 9, 13, 15, 0),
|
|
"to": datetime(2025, 9, 14, 12, 0),
|
|
"latitude": 51.8745678,
|
|
"longitude": -0.3844256,
|
|
},
|
|
{
|
|
"location": "Akranes",
|
|
"country": "is",
|
|
"from": datetime(2025, 9, 14, 16, 0),
|
|
"to": datetime(2025, 9, 16, 11, 0),
|
|
"latitude": 64.4344723,
|
|
"longitude": -21.5517961,
|
|
},
|
|
{
|
|
"location": "Ísafjörður",
|
|
"country": "is",
|
|
"from": datetime(2025, 9, 16, 15, 0),
|
|
"to": datetime(2025, 10, 1, 11, 0),
|
|
"latitude": 66.072445,
|
|
"longitude": -23.120111,
|
|
},
|
|
],
|
|
)
|
|
|
|
journeys = agenda.car_journey_yaml.car_journeys_for_trip(
|
|
trip, tmp_path, destination="fly-drive"
|
|
)
|
|
|
|
assert [journey.route for journey in journeys] == [
|
|
"PCH_to_Luton_Airport",
|
|
"KEF_to_Akranes",
|
|
"Akranes_to_Isafjordur",
|
|
"Isafjordur_to_KEF",
|
|
"LTN_to_PCH",
|
|
]
|
|
assert [journey.depart for journey in journeys] == [
|
|
date(2025, 9, 13),
|
|
date(2025, 9, 14),
|
|
date(2025, 9, 16),
|
|
date(2025, 10, 1),
|
|
date(2025, 10, 1),
|
|
]
|
|
|
|
|
|
def test_write_route_files_skips_existing_routes(tmp_path: Path) -> None:
|
|
"""Existing route files should not call the route API."""
|
|
write_home_route(tmp_path)
|
|
journey = agenda.car_journey_yaml.CarJourney(
|
|
trip=date(2025, 4, 22),
|
|
depart=date(2025, 4, 22),
|
|
arrive=date(2025, 4, 22),
|
|
route="PCH_to_EMF",
|
|
start=(-2.60283, 51.44083),
|
|
end=(-2.37789, 52.038),
|
|
)
|
|
|
|
def fail_fetch(
|
|
_start: agenda.car_journey_yaml.LonLat,
|
|
_end: agenda.car_journey_yaml.LonLat,
|
|
) -> agenda.car_journey_yaml.GeoJSON:
|
|
raise AssertionError("route API should not be called")
|
|
|
|
written = agenda.car_journey_yaml.write_route_files(tmp_path, [journey], fail_fetch)
|
|
|
|
assert written == 0
|
|
|
|
|
|
def test_import_car_journeys_inserts_chronologically_and_is_idempotent(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Generated journeys should be inserted without duplicates."""
|
|
(tmp_path / "car_journeys.yaml").write_text("""---
|
|
- trip: 2025-05-01
|
|
depart: 2025-05-01
|
|
arrive: 2025-05-01
|
|
route: PCH_to_Later
|
|
""")
|
|
journeys = [
|
|
agenda.car_journey_yaml.CarJourney(
|
|
trip=date(2025, 4, 22),
|
|
depart=date(2025, 4, 22),
|
|
arrive=date(2025, 4, 22),
|
|
route="PCH_to_Callestick",
|
|
start=(-2.60283, 51.44083),
|
|
end=(-5.13466, 50.307093),
|
|
),
|
|
agenda.car_journey_yaml.CarJourney(
|
|
trip=date(2025, 4, 22),
|
|
depart=date(2025, 4, 25),
|
|
arrive=date(2025, 4, 25),
|
|
route="Callestick_to_PCH",
|
|
start=(-5.13466, 50.307093),
|
|
end=(-2.60283, 51.44083),
|
|
),
|
|
]
|
|
|
|
added = agenda.car_journey_yaml.import_car_journeys(tmp_path, journeys)
|
|
added_again = agenda.car_journey_yaml.import_car_journeys(tmp_path, journeys)
|
|
|
|
assert added == 2
|
|
assert added_again == 0
|
|
text = (tmp_path / "car_journeys.yaml").read_text()
|
|
assert text.index("PCH_to_Callestick") < text.index("PCH_to_Later")
|