def test_post_sanitize_header_pc():
test_subject = FITSHelper(io.BytesIO(), io.BytesIO())
data = np.arange(10000).reshape(100, 100)
header = Header()
wcs = WCS(fix=False)
wcs.wcs.pc = [[0.9, 0.8], [0.7, 0.6]]
header.set('REMAIN1', 'VALUE1')
header.set('DQ1', 'dqvalue1')
header.set('NAXIS', 2)
header.set('NAXIS1', 88)
header.set('NAXIS2', 212)
header.set('CRPIX1', 77.0)
header.set('WCSAXES', 2)
header.set('CTYPE1', 'ctype1value')
assert header.get('PC1_1') is None, 'Should not contain PC1_1'
assert header.get('PC2_1') is None, 'Should not contain PC2_1'
assert header.get('PC1_2') is None, 'Should not contain PC1_2'
assert header.get('PC2_2') is None, 'Should not contain PC2_2'
result = CutoutResult(data, wcs=wcs)
assert header.index('WCSAXES') > header.index('CRPIX1'), \
'Start with bad indexes...'
test_subject._post_sanitize_header(header, result)
assert header.get('PC1_1') == 0.9, 'Wrong PC1_1 value.'
assert header.get('PC2_1') == 0.7, 'Wrong PC2_1 value.'
assert header.get('PC1_2') == 0.8, 'Wrong PC1_2 value.'
assert header.get('PC2_2') == 0.6, 'Wrong PC2_2 value.'
assert 'VALUE1' == header.get('REMAIN1'), 'REMAIN1 should still be there.'
def test__check_hdu_list():
in_io = io.BytesIO(b'TEST STRING HERE')
out_io = io.BytesIO()
test_subject = FITSHelper(in_io, out_io)
dimensions = [PixelCutoutHDU([(20, 35), (40, 50)], extension=1)]
hdu_list = _create_hdu_list()
assert test_subject._check_hdu_list(dimensions, hdu_list), 'Should match.'
def test_is_extension_requested():
test_subject = FITSHelper(io.BytesIO(), io.BytesIO())
dimension = PixelCutoutHDU([(400, 800)], extension=1)
assert not test_subject._is_extension_requested('4', ('EXN', 4), dimension)
dimension = PixelCutoutHDU([(400, 800)])
assert not test_subject._is_extension_requested('4', ('NOM', 1), dimension)
dimension = PixelCutoutHDU([(400, 800)], extension=2)
assert test_subject._is_extension_requested('2', ('NOM', 7), dimension)
def test__check_hdu_list():
in_io = io.BytesIO(b'TEST STRING HERE')
out_io = io.BytesIO()
test_subject = FITSHelper(in_io, out_io)
dimensions = [PixelCutoutHDU([(20, 35), (40, 50)], extension=1)]
hdu_list = _create_hdu_list()
has_match = test_subject._check_hdu_list(dimensions, hdu_list)
logging.debug('Output from _check_hdu_list() is {}'.format(has_match))
assert has_match, 'Should match.'
def test_post_sanitize_header():
test_subject = FITSHelper(io.BytesIO(), io.BytesIO())
data = np.arange(10000).reshape(100, 100)
header = Header()
wcs = WCS(fix=False)
header.set('REMAIN1', 'VALUE1')
header.set('DQ1', 'dqvalue1')
header.set('NAXIS', 2)
header.set('NAXIS1', 88)
header.set('NAXIS2', 212)
result = CutoutResult(data, wcs=wcs)
test_subject._post_sanitize_header(header, result)
assert 'VALUE1' == header.get('REMAIN1'), 'REMAIN1 should still be there.'
def test_cd_pc_header_fix():
test_subject = FITSHelper(io.BytesIO(), io.BytesIO())
data = np.arange(10000).reshape(100, 100)
header = Header()
wcs = WCS()
header.set('NAXIS', 2)
header.set('NAXIS1', 88)
header.set('NAXIS2', 212)
header.set('CD1_1', 44)
header.set('CD1_2', 88)
header.set('CD2_1', 22)
header.set('CD2_2', 33)
result = CutoutResult(data, wcs=wcs)
test_subject._post_sanitize_header(header, result)
assert not header.get('CD1_2'), 'CD1_2 should be renamed.'
assert not header.get('CD1_1'), 'CD1_1 should be renamed.'
assert 44 == header.get('PC1_1'), 'PC1_1 should be 44.'
assert 33 == header.get('PC2_2'), 'PC2_2 should be 33.'
def test_pc_leading_zeroes_header_fix():
test_subject = FITSHelper(io.BytesIO(), io.BytesIO())
data = np.arange(10000).reshape(100, 100)
header = Header()
wcs = WCS()
header.set('NAXIS', 2)
header.set('NAXIS1', 88)
header.set('NAXIS2', 212)
header.set('PC01_01', 44)
header.set('PC01_02', 88)
header.set('PC02_01', 22)
header.set('PC02_02', 33)
result = CutoutResult(data, wcs=wcs)
test_subject._post_sanitize_header(header, result)
assert not header.get('PC01_02'), 'PC01_02 should be renamed.'
assert not header.get('PC01_01'), 'PC01_01 should be renamed.'
assert 22 == header.get('PC2_1'), 'PC2_1 should be 22.'
assert 88 == header.get('PC1_2'), 'PC1_2 should be 88.'
def test_post_sanitize_header_crpix():
test_subject = FITSHelper(io.BytesIO(), io.BytesIO())
data = np.arange(10000).reshape(100, 100)
header = Header()
wcs = WCS(fix=False)
header.set('REMAIN1', 'VALUE1')
header.set('DQ1', 'dqvalue1')
header.set('NAXIS', 2)
header.set('NAXIS1', 88)
header.set('NAXIS2', 212)
header.set('CRPIX1', 77.0)
header.set('WCSAXES', 2)
header.set('CTYPE1', 'ctype1value')
result = CutoutResult(data, wcs=wcs)
assert header.index('WCSAXES') > header.index('CRPIX1'), \
'Start with bad indexes...'
test_subject._post_sanitize_header(header, result)
assert 'VALUE1' == header.get('REMAIN1'), 'REMAIN1 should still be there.'
