[TESTED] Updated ForestModelLoadAndSave.py

Forst model specific values added in test setup for random Trips
e-mission · Mar 22, 2024 · 450094c · 450094c
1 parent 19bb394
commit 450094c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/emission/tests/modellingTests/TestForestModelLoadandSave.py b/emission/tests/modellingTests/TestForestModelLoadandSave.py
@@ -67,6 +67,10 @@ def setUp(self):
                 has_label_p=self.has_label_percent
             )
 
+            for result_entry in test_data:
+                result_entry['data']['start_local_dt']=result_entry['metadata']['write_local_dt']
+                result_entry['data']['end_local_dt']=result_entry['metadata']['write_local_dt']
+
             ts.bulk_insert(test_data)
 
             # confirm data write did not fail