Use datetime for ramp-up calculations

1kastner · May 23, 2024 · 25c1427 · 25c1427
1 parent 30760e5
commit 25c1427
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/conflowgen/flow_generator/container_flow_generation_service.py b/conflowgen/flow_generator/container_flow_generation_service.py
@@ -45,8 +45,12 @@ def _update_generation_properties_and_distributions(self):
 
         ramp_up_period = container_flow_generation_properties.ramp_up_period
         ramp_down_period = container_flow_generation_properties.ramp_down_period
-        self.ramp_up_period_end = self.container_flow_start_date + datetime.timedelta(days=ramp_up_period)
-        self.ramp_down_period_start = self.container_flow_end_date - datetime.timedelta(days=ramp_down_period)
+        self.ramp_up_period_end = datetime.datetime.combine(
+            self.container_flow_start_date, datetime.time(hour=0, minute=0, second=0)
+        ) + datetime.timedelta(days=ramp_up_period)
+        self.ramp_down_period_start = datetime.datetime.combine(
+            self.container_flow_end_date, datetime.time(hour=0, minute=0, second=0)
+        ) - datetime.timedelta(days=ramp_down_period)
         assert self.ramp_up_period_end <= self.ramp_down_period_start
 
         self.transportation_buffer: float = container_flow_generation_properties.transportation_buffer