From 19bb394c8b2506036102f8008b05648e3c4c0ebb Mon Sep 17 00:00:00 2001 From: $aTyam Date: Fri, 22 Mar 2024 03:16:35 -0400 Subject: [PATCH] Update TestForestModelIntegration.py 2 more ( total 4) Forest model specific features are now added after generating random trips for testing purpose. --- .../tests/modellingTests/TestForestModelIntegration.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/emission/tests/modellingTests/TestForestModelIntegration.py b/emission/tests/modellingTests/TestForestModelIntegration.py index 44fe85afe..88813b5c4 100644 --- a/emission/tests/modellingTests/TestForestModelIntegration.py +++ b/emission/tests/modellingTests/TestForestModelIntegration.py @@ -12,6 +12,8 @@ import emission.tests.common as etc import emission.pipeline.intake_stage as epi import logging +from bson.objectid import ObjectId + import emission.analysis.modelling.trip_model.config as eamtc import emission.analysis.modelling.trip_model.run_model as eamur @@ -54,11 +56,12 @@ def setUp(self): threshold=0.004, # ~400m has_label_p=0.9 ) - + ## Required for Forest model inference for result_entry in train: result_entry['data']['start_local_dt']=result_entry['metadata']['write_local_dt'] result_entry['data']['end_local_dt']=result_entry['metadata']['write_local_dt'] - + result_entry['data']['start_place']=ObjectId() + result_entry['data']['end_place']=ObjectId() ts.bulk_insert(train) # confirm data write did not fail check_data = esda.get_entries(key="analysis/confirmed_trip", user_id=self.testUUID, time_query=None)