def test_passes_if_header_does_have_all_keys(self):
hdr = wcs.WCS().to_header()
hdr["NAXIS1"] = 100
assert has_needed_keywords(hdr)
def test_passes_if_header_does_have_all_keys_and_suffix(self):
hdr = wcs.WCS(key="D").to_header()
hdr["NAXIS1"] = 100
assert has_needed_keywords(hdr, "D")
def test_fails_if_header_does_not_have_all_keys(self):
assert not has_needed_keywords(fits.Header())