Skip to content

Commit dc26d7e

Browse files
committed
simplify test configs
1 parent b37f74b commit dc26d7e

File tree

1 file changed

+32
-16
lines changed

1 file changed

+32
-16
lines changed

test/test_mtc.py

Lines changed: 32 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ def _test_path(dirname):
2424
def 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

182181
EXPECTED_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")
226225
def 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

Comments
 (0)