From cbcd81b16f0ed2644edec412c3790dae6dea916e Mon Sep 17 00:00:00 2001 From: Jeremy Kubica <104161096+jeremykubica@users.noreply.github.com> Date: Fri, 12 Jan 2024 16:04:31 -0500 Subject: [PATCH] Address PR comment --- tests/test_wcs_utils.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/tests/test_wcs_utils.py b/tests/test_wcs_utils.py index 628a701c..e12d9937 100644 --- a/tests/test_wcs_utils.py +++ b/tests/test_wcs_utils.py @@ -51,18 +51,18 @@ def test_append_wcs_to_hdu_header(self): else: wcs_info = self.wcs - # Run the dictionary and full WCS tests as separate subtests - with self.subTest(i=use_dictionary): - pri = fits.PrimaryHDU() - self.assertFalse("CRVAL1" in pri.header) - self.assertFalse("CRVAL2" in pri.header) - self.assertFalse("CRPIX1" in pri.header) - self.assertFalse("CRPIX2" in pri.header) - - append_wcs_to_hdu_header(wcs_info, pri.header) - for key in self.header_dict: - self.assertTrue(key in pri.header) - self.assertAlmostEqual(pri.header[key], self.header_dict[key]) + # Run the dictionary and full WCS tests as separate subtests + with self.subTest(i=use_dictionary): + pri = fits.PrimaryHDU() + self.assertFalse("CRVAL1" in pri.header) + self.assertFalse("CRVAL2" in pri.header) + self.assertFalse("CRPIX1" in pri.header) + self.assertFalse("CRPIX2" in pri.header) + + append_wcs_to_hdu_header(wcs_info, pri.header) + for key in self.header_dict: + self.assertTrue(key in pri.header) + self.assertAlmostEqual(pri.header[key], self.header_dict[key]) def test_make_fake_wcs_info(self): # Test that we make the dictionary