def test_header_value_negative(self, kw):
hdu = fits.CompImageHDU(np.ones((10, 10)))
hdu._header[kw] = -1
with pytest.raises(ValueError) as exc:
compress_hdu(hdu)
assert '{} should not be negative.'.format(kw) in str(exc)
def test_header_value_longlong_overflow(self, kw):
hdu = fits.CompImageHDU(np.ones((10, 10)))
hdu._header[kw] = MAX_LONGLONG + 1
with pytest.raises(OverflowError):
compress_hdu(hdu)
def test_header_value_float_overflow(self, kw):
hdu = fits.CompImageHDU(np.ones((10, 10)))
hdu._header[kw] = 1e300
with pytest.raises(OverflowError):
compress_hdu(hdu)
def test_invalid_zdither(self):
hdu = fits.CompImageHDU(np.ones((10, 10)), quantize_method=1)
hdu._header['ZDITHER0'] = 'a'
with pytest.raises(TypeError):
compress_hdu(hdu)
def test_header_missing_keyword(self, kw):
hdu = fits.CompImageHDU(np.ones((10, 10)))
del hdu._header[kw]
with pytest.raises(KeyError) as exc:
compress_hdu(hdu)
assert kw in str(exc)
def test_missing_internal_header(self):
hdu = fits.CompImageHDU(np.ones((10, 10)))
del hdu._header
with pytest.raises(AttributeError) as exc:
compress_hdu(hdu)
assert '_header' in str(exc)
def test_invalid_tform(self):
hdu = fits.CompImageHDU(np.ones((10, 10)))
hdu._header['TFORM1'] = 'TX'
with pytest.raises(RuntimeError) as exc:
compress_hdu(hdu)
assert 'TX' in str(exc) and 'TFORM' in str(exc)
def test_zbitpix_unknown(self):
hdu = fits.CompImageHDU(np.ones((10, 10)))
hdu._header['ZBITPIX'] = 13
with pytest.raises(ValueError) as exc:
compress_hdu(hdu)
assert 'Invalid value for BITPIX: 13' in str(exc)
def test_data_none(self):
hdu = fits.CompImageHDU(np.ones((10, 10)))
hdu.data = None
with pytest.raises(TypeError) as exc:
compress_hdu(hdu)
assert 'CompImageHDU.data must be a numpy.ndarray' in str(exc)
def test_unknown_compression_type(self):
hdu = fits.CompImageHDU(np.ones((10, 10)))
hdu._header['ZCMPTYPE'] = 'fun'
with pytest.raises(ValueError) as exc:
compress_hdu(hdu)
assert 'Unrecognized compression type: fun' in str(exc)
def test_wrong_argument_number(self):
with pytest.raises(TypeError):
compress_hdu(1, 2)
def test_header_value_no_double_int_image(self, kw):
hdu = fits.CompImageHDU(np.ones((10, 10), dtype=np.int32))
hdu._header[kw] = '1'
with pytest.raises(TypeError):
compress_hdu(hdu)
def test_header_value_no_string(self, kw):
hdu = fits.CompImageHDU(np.ones((10, 10)))
hdu._header[kw] = 1
with pytest.raises(TypeError):
compress_hdu(hdu)
def test_header_value_exceeds_custom_limit(self, kw, limit):
hdu = fits.CompImageHDU(np.ones((10, 10)))
hdu._header[kw] = limit + 1
with pytest.raises(ValueError) as exc:
compress_hdu(hdu)
assert kw in str(exc)