diff --git a/test/regress/reference-pipeline-hh100-recode.parquetpipeline/checkpoints.parquet b/test/regress/reference-pipeline-hh100-recode.parquetpipeline/checkpoints.parquet index f0e7e8b..bb92a81 100644 Binary files a/test/regress/reference-pipeline-hh100-recode.parquetpipeline/checkpoints.parquet and b/test/regress/reference-pipeline-hh100-recode.parquetpipeline/checkpoints.parquet differ diff --git a/test/regress/reference-pipeline-hh100-recode.parquetpipeline/land_use_taz/initialize_landuse_taz.parquet b/test/regress/reference-pipeline-hh100-recode.parquetpipeline/land_use_taz/initialize_landuse_taz.parquet new file mode 100644 index 0000000..5d8f918 Binary files /dev/null and b/test/regress/reference-pipeline-hh100-recode.parquetpipeline/land_use_taz/initialize_landuse_taz.parquet differ diff --git a/test/regress/reference-pipeline-hh100.parquetpipeline/checkpoints.parquet b/test/regress/reference-pipeline-hh100.parquetpipeline/checkpoints.parquet index cfdc780..448287a 100644 Binary files a/test/regress/reference-pipeline-hh100.parquetpipeline/checkpoints.parquet and b/test/regress/reference-pipeline-hh100.parquetpipeline/checkpoints.parquet differ diff --git a/test/regress/reference-pipeline-hh100.parquetpipeline/land_use_taz/initialize_landuse_taz.parquet b/test/regress/reference-pipeline-hh100.parquetpipeline/land_use_taz/initialize_landuse_taz.parquet new file mode 100644 index 0000000..2758487 Binary files /dev/null and b/test/regress/reference-pipeline-hh100.parquetpipeline/land_use_taz/initialize_landuse_taz.parquet differ diff --git a/test/test_sandag_abm3.py b/test/test_sandag_abm3.py index 0ea4e9f..81543d6 100644 --- a/test/test_sandag_abm3.py +++ b/test/test_sandag_abm3.py @@ -42,6 +42,7 @@ def regress(out_dir: Path, regress_dir: Path = None): "initialize_proto_population", "compute_disaggregate_accessibility", "initialize_landuse", + "initialize_landuse_taz", "initialize_households", "compute_accessibility", "av_ownership", @@ -135,7 +136,7 @@ def test_sandag_abm3_progressive(use_sharrow): tags = "".join(tags) ref_pipeline = Path(__file__).parent.joinpath( - f"regress/reference-pipeline{tags}.zip" + f"regress/reference-pipeline{tags}.parquetpipeline" ) for step_name in EXPECTED_MODELS: @@ -156,9 +157,14 @@ def test_sandag_abm3_progressive(use_sharrow): # make new reference pipeline file if it is missing import shutil - shutil.make_archive( - ref_pipeline.with_suffix(""), "zip", state.checkpoint.store.filename - ) + if ref_pipeline.suffix == ".zip": + shutil.make_archive( + ref_pipeline.with_suffix(""), "zip", state.checkpoint.store.filename + ) + else: + shutil.copytree(state.checkpoint.store.filename, ref_pipeline) + if (ref_pipeline / ".gitignore").exists(): + os.remove(ref_pipeline / ".gitignore") if __name__ == "__main__":