From 58216a57a211db2ba9d06453cb5c57fed15dad48 Mon Sep 17 00:00:00 2001 From: "Brandon N. Benton" Date: Wed, 25 Sep 2024 14:07:15 -0600 Subject: [PATCH] fixed pipeline with mask test --- tests/pipeline/test_pipeline.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/pipeline/test_pipeline.py b/tests/pipeline/test_pipeline.py index 68bc1fb6f..988709c38 100644 --- a/tests/pipeline/test_pipeline.py +++ b/tests/pipeline/test_pipeline.py @@ -362,7 +362,11 @@ def test_fwp_pipeline_with_mask(input_files): assert os.path.exists(fp_out) with ResourceX(fp_out) as f: assert len(f.time_index) == t_enhance * n_tsteps - assert len(f.meta) == s_enhance * s_enhance * np.prod(shape) + + # unmasked gives 4 chunks so without chunk index 2 we have just 3 + assert len(f.meta) == s_enhance * s_enhance * 3 * np.prod( + fp_chunk_shape[:2] + ) status_fps = glob.glob(f'{td}/.gaps/*status*.json') assert len(status_fps) == 1