@@ -24,7 +24,6 @@ def _test_path(dirname):
2424def run_test_mtc (
2525 multiprocess = False , chunkless = False , recode = False , sharrow = False , extended = False
2626):
27-
2827 def regress (ext ):
2928 if ext :
3029 regress_trips_df = pd .read_csv (_test_path ("regress/final_trips-ext.csv" ))
@@ -180,9 +179,9 @@ def test_mtc_sharrow_ext():
180179
181180
182181EXPECTED_MODELS = [
183- ' input_checker' ,
184- ' initialize_proto_population' ,
185- ' compute_disaggregate_accessibility' ,
182+ " input_checker" ,
183+ " initialize_proto_population" ,
184+ " compute_disaggregate_accessibility" ,
186185 "initialize_landuse" ,
187186 "initialize_households" ,
188187 "compute_accessibility" ,
@@ -224,29 +223,46 @@ def test_mtc_sharrow_ext():
224223
225224@testing .run_if_exists ("reference-pipeline-extended.zip" )
226225def test_mtc_extended_progressive ():
227-
228226 import activitysim .abm # register components # noqa: F401
229227
230228 out_dir = _test_path ("output-progressive" )
231229 Path (out_dir ).mkdir (exist_ok = True )
232230 Path (out_dir ).joinpath (".gitignore" ).write_text ("**\n " )
233231
234232 working_dir = Path (_example_path ("." ))
235- data_dir = _example_path ("data" )
236- data_model_dir = _example_path ("data_model" )
233+
234+ settings = {
235+ "treat_warnings_as_errors" : False ,
236+ "households_sample_size" : 10 ,
237+ "chunk_size" : 0 ,
238+ "use_shadow_pricing" : False ,
239+ "want_dest_choice_sample_tables" : False ,
240+ "cleanup_pipeline_after_run" : True ,
241+ "output_tables" : {
242+ "h5_store" : False ,
243+ "action" : "include" ,
244+ "prefix" : "final_" ,
245+ "sort" : True ,
246+ "tables" : [
247+ {
248+ "tablename" : "trips" ,
249+ "decode_columns" : {
250+ "origin" : "land_use.zone_id" ,
251+ "destination" : "land_use.zone_id" ,
252+ },
253+ },
254+ ],
255+ },
256+ "recode_pipeline_columns" : True ,
257+ }
237258
238259 state = workflow .State .make_default (
239260 working_dir = working_dir ,
240- configs_dir = (
241- _test_path ("configs_recode" ),
242- _test_path ("ext-configs" ),
243- _example_path ("ext-configs" ),
244- _test_path ("configs" ),
245- _example_path ("configs" ),
246- ),
247- data_dir = data_dir ,
248- data_model_dir = data_model_dir ,
261+ configs_dir = ("ext-configs" , "configs" ),
262+ data_dir = "data" ,
263+ data_model_dir = "data_model" ,
249264 output_dir = out_dir ,
265+ settings = settings ,
250266 )
251267 state .filesystem .persist_sharrow_cache ()
252268
0 commit comments