From ebfdbe5399c3e11aa56b5d0ecd3be3976a405028 Mon Sep 17 00:00:00 2001 From: Drew Leonard Date: Wed, 8 Nov 2023 14:35:50 +0000 Subject: [PATCH] Check Stokes headers of Stokes-cropped datasets --- dkist/dataset/tests/test_crop_dataset.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/dkist/dataset/tests/test_crop_dataset.py b/dkist/dataset/tests/test_crop_dataset.py index 4b0ff6ef..f61dfec9 100644 --- a/dkist/dataset/tests/test_crop_dataset.py +++ b/dkist/dataset/tests/test_crop_dataset.py @@ -19,8 +19,7 @@ def test_crop_visp_by_only_stokes(croppable_visp_dataset): assert cropped.wcs.pixel_n_dim == croppable_visp_dataset.wcs.pixel_n_dim - 1 assert cropped.data.shape == croppable_visp_dataset.data.shape[1:] - # This won't be true yet because of bugs in the Stokes WCS - # assert (cropped.headers['DINDEX4'] == 1).all() + assert (cropped.headers['DINDEX4'] == 1).all() def test_crop_visp_by_time(croppable_visp_dataset): @@ -107,8 +106,7 @@ def test_crop_cryo_by_only_stokes(croppable_cryo_dataset): assert cropped.wcs.pixel_n_dim == croppable_cryo_dataset.wcs.pixel_n_dim - 1 assert cropped.data.shape == croppable_cryo_dataset.data.shape[1:] - # This won't be true yet because of bugs in the Stokes WCS - # assert (cropped.headers['DINDEX4'] == 1).all() + assert (cropped.headers['DINDEX4'] == 1).all() def test_crop_cryo_by_time(croppable_cryo_dataset):