diff --git a/test/test_models.py b/test/test_models.py index 42f805f..6dff3aa 100644 --- a/test/test_models.py +++ b/test/test_models.py @@ -1,4 +1,5 @@ import pytest +import datetime from app.model.models import ( UserBase, UserResponse, UserCreate, UserDB, UserUpdate, GroupBase, GroupCreate, GroupDB, GroupResponse, @@ -68,13 +69,13 @@ def test_timetable_models(): # Test TimetableBase with valid values timetable = TimetableCreate(weekday=1, starttime="09:00", duration=120) assert timetable.weekday == 1 - assert timetable.starttime == "09:00" + assert timetable.starttime == datetime.time(9, 0) assert timetable.duration == 120 # Test boundary values - max_duration = TimetableCreate(weekday=7, starttime="23:59", duration=1439) + max_duration = TimetableCreate(weekday=6, starttime="23:59", duration=1439) assert max_duration.duration == 1439 - assert max_duration.weekday == 7 + assert max_duration.weekday == 6 def test_token_models():