Skip to content

Commit

Permalink
Address PR comment
Browse files Browse the repository at this point in the history
  • Loading branch information
jeremykubica committed Jan 12, 2024
1 parent 049a544 commit cbcd81b
Showing 1 changed file with 12 additions and 12 deletions.
24 changes: 12 additions & 12 deletions tests/test_wcs_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit cbcd81b

Please sign in to comment.