SIENTIAPDE-1193

Enhance test cases to include datetime_columns for improved timestamp handling in minimal_retrain and predictions_batch functions

tests/orchestrator/utils/test_orchestrator_functions.py

@@ -41,6 +41,7 @@ def test_minimal_retrain():
             "name": "test_model_name"
         },
         "query": "select * from sientia_data.laborious_data order by \"timestamp\" desc limit 30;",
+        "datetime_columns": ["timestamp"]
     }
     result = minimal_retrain(config)
     expected = {
@@ -53,6 +54,7 @@ def test_minimal_retrain():
         "query": "select * from sientia_data.laborious_data order by \"timestamp\" desc limit 30;",
         "schema": "sientia_data",
         "table_name": "log_retrain",
+        "datetime_columns": ["timestamp"]
     }
     assert result == expected
 
@@ -195,6 +197,7 @@ def test_predictions_batch(mock_process_path_priority,
             }
         ],
         "path_priority": ["STOP", "CONTINUE", "REPEAT"],
+        "datetime_columns": ["timestamp"]
     }
 
     result = predictions_batch(config)
@@ -268,7 +271,8 @@ def test_predictions_batch(mock_process_path_priority,
                 "config": {}
             }
         },
-        "path_priority": ["STOP", "CONTINUE", "REPEAT"]
+        "path_priority": ["STOP", "CONTINUE", "REPEAT"],
+        "datetime_columns": ["timestamp"]
     }
     assert result == expected