edward/agenda
1
0
Fork 0
Code Issues 16 Pull requests Packages Projects Releases Wiki Activity

More tests

Browse source
This commit is contained in:
Edward Betts 2025-07-16 06:51:18 +02:00
parent 9dd033ba77
commit 3ad11b070a
1 changed files with 14 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tests/test_busy.py
View file

@ -1,4 +1,4 @@
from datetime import date, datetime, timedelta
from datetime import date, datetime
import agenda.busy
import agenda.travel as travel
@ -10,22 +10,25 @@ def test_get_location_for_date() -> None:
app.config["SERVER_NAME"] = "test"
with app.app_context():
today = datetime.now().date()
start = date(2023, 1, 1)
trips = agenda.trip.build_trip_list()
data_dir = app.config["PERSONAL_DATA"]
busy_events = agenda.busy.get_busy_events(start, app.config, trips)
weekends = agenda.busy.weekends(start, busy_events, trips, data_dir)
# Parse YAML files once for the test
bookings = travel.parse_yaml("flights", data_dir)
accommodations = travel.parse_yaml("accommodation", data_dir)
airports = travel.parse_yaml("airports", data_dir)
for year in range(2023, 2025):
start = date(2023, 1, 1)
busy_events = agenda.busy.get_busy_events(start, app.config, trips)
weekends = agenda.busy.weekends(start, busy_events, trips, data_dir)
for weekend in weekends:
for day in "saturday", "sunday":
assert weekend[day + "_location"][0] == "Bristol" or bool(weekend[day])
assert weekend[day + "_location"][0] == "Bristol" or bool(
weekend[day]
)
# Debug the April 29 issue
april_29_location = agenda.busy.get_location_for_date(