def test_sorting(self, triage_setup):
collection = ImageFileCollection(
location=triage_setup.test_dir,
keywords=['imagetyp', 'filter', 'object'])
all_elements = []
for hdu, fname in collection.hdus(return_fname=True):
all_elements.append((str(hdu.header), fname))
# Now sort
collection.sort(keys=['imagetyp', 'object'])
# and check it's all still right
for hdu, fname in collection.hdus(return_fname=True):
assert((str(hdu.header), fname) in all_elements)
for i in range(len(collection.summary)):
assert(collection.summary['file'][i] == collection.files[i])
def test_hdus_masking(self, triage_setup):
collection = ImageFileCollection(location=triage_setup.test_dir,
keywords=['imagetyp', 'exposure'])
old_data = np.array(collection.summary)
for hdu in collection.hdus(imagetyp='bias'):
pass
new_data = np.array(collection.summary)
assert (new_data == old_data).all()
def test_hdus(self, triage_setup):
collection = ImageFileCollection(
location=triage_setup.test_dir, keywords=['imagetyp'])
n_hdus = 0
for hdu in collection.hdus():
assert isinstance(hdu, fits.PrimaryHDU)
data = hdu.data # must access the data to force scaling
# pre-astropy 1.1 unsigned data was changed to float32 and BZERO
# removed. In 1.1 and later, BZERO stays but the data type is
# unsigned int.
assert (('BZERO' not in hdu.header) or
(data.dtype is np.dtype(np.uint16)))
n_hdus += 1
assert n_hdus == triage_setup.n_test['files']
def test_multiple_extensions(self, triage_setup, extension):
ext1 = fits.PrimaryHDU()
ext1.data = np.arange(1, 5)
# It is important than the name used for this test extension
# NOT be MASK or UNCERT because both are treated in a special
# way by the FITS reader.
test_ext_name = 'TESTEXT'
ext2 = fits.ImageHDU(name=test_ext_name)
ext2.data = np.arange(6, 10)
hdulist = fits.hdu.hdulist.HDUList([ext1, ext2])
hdulist.writeto(os.path.join(triage_setup.test_dir,
'multi-extension.fits'))
ic2 = ImageFileCollection(
triage_setup.test_dir, keywords='*',
filenames=['multi-extension.fits'], ext=extension)
ic1 = ImageFileCollection(
triage_setup.test_dir,
keywords='*', filenames=['multi-extension.fits'], ext=0)
assert ic1.ext == 0
assert ic2.ext == extension
column2 = ic2.summary.colnames
column1 = ic1.summary.colnames
assert column1 != column2
list1 = [key.lower() for key in ext2.header]
list2 = ic2.summary.colnames[1:]
assert list1 == list2
ccd_kwargs = {'unit': 'adu'}
for data, hdr, hdu, ccd in zip(ic2.data(),
ic2.headers(),
ic2.hdus(),
ic2.ccds(ccd_kwargs)):
np.testing.assert_array_equal(data, ext2.data)
assert hdr == ext2.header
# Now compare that the generators each give the same stuff
np.testing.assert_array_equal(data, ccd.data)
np.testing.assert_array_equal(data, hdu.data)
assert hdr == hdu.header
assert hdr == ccd.meta
