11import datetime
22
3+ from shepherd_core import local_tz
34from shepherd_core .data_models .experiment import Experiment
45from shepherd_server .api_experiment .models import WebExperiment
56from shepherd_server .api_user .models import User
@@ -19,7 +20,8 @@ async def test_get_next_scheduling(
1920
2021 assert await WebExperiment .get_next_scheduling () is None
2122
22- one .requested_execution_at = datetime .date (2000 , 1 , 1 )
23+ tzone = local_tz ()
24+ one .requested_execution_at = datetime .datetime (2000 , 1 , 1 , tzinfo = tzone )
2325 await one .save_changes ()
2426
2527 _next = await WebExperiment .get_next_scheduling ()
@@ -30,19 +32,19 @@ async def test_get_next_scheduling(
3032
3133 _next = await WebExperiment .get_next_scheduling ()
3234 assert _next .id == one .id
33- two .requested_execution_at = datetime .date (2001 , 1 , 1 )
35+ two .requested_execution_at = datetime .datetime (2001 , 1 , 1 , tzinfo = tzone )
3436 await two .save_changes ()
3537
3638 _next = await WebExperiment .get_next_scheduling ()
3739 assert _next .id == one .id
3840
39- two .requested_execution_at = datetime .date (1999 , 1 , 1 )
41+ two .requested_execution_at = datetime .datetime (1999 , 1 , 1 , tzinfo = tzone )
4042 await two .save_changes ()
4143
4244 _next = await WebExperiment .get_next_scheduling ()
4345 assert _next .id == two .id
4446
45- two .started_at = datetime .date (1999 , 1 , 1 )
47+ two .started_at = datetime .datetime (1999 , 1 , 1 , tzinfo = tzone )
4648 await two .save_changes ()
4749
4850 _next = await WebExperiment .get_next_scheduling ()
0 commit comments