def test_io(sobj1, sobj2, sobj3, sobj4):
sobjs = specobjs.SpecObjs([sobj1, sobj2, sobj3, sobj4])
sobjs[0]['BOX_WAVE'] = np.arange(1000).astype(float)
sobjs[1]['BOX_WAVE'] = np.arange(1000).astype(float)
sobjs[2]['BOX_WAVE'] = np.arange(1000).astype(float)
#sobjs[0]['BOX_COUNTS'] = np.ones_like(sobjs[0].BOX_WAVE) # This tests single array
sobjs[1]['BOX_COUNTS'] = np.ones_like(sobjs[0].BOX_WAVE)
sobjs[2]['BOX_COUNTS'] = np.ones_like(sobjs[0].BOX_WAVE)
# Detector
sobjs[0]['DETECTOR'] = tstutils.get_kastb_detector()
tmp = tstutils.get_kastb_detector()
tmp['det'] = 2
sobjs[1]['DETECTOR'] = tmp
# Write
header = fits.PrimaryHDU().header
header['TST'] = 'TEST'
ofile = data_path('tst_specobjs.fits')
if os.path.isfile(ofile):
os.remove(ofile)
sobjs.write_to_fits(header, ofile, overwrite=False)
# Read
hdul = fits.open(ofile)
assert len(hdul) == 7 # Primary + 4 Obj + 2 Detectors
assert hdul[0].header['NSPEC'] == 4
hdul.close()
#
_sobjs = specobjs.SpecObjs.from_fitsfile(ofile)
assert _sobjs.nobj == 4
assert np.array_equal(sobjs[0].BOX_WAVE, _sobjs[0].BOX_WAVE)
assert np.array_equal(sobjs[1].BOX_WAVE, _sobjs[1].BOX_WAVE)
_sobjs.write_to_fits(header, ofile, overwrite=True)
# Detector
assert _sobjs[0].DETECTOR is not None, '1st object started with Detector'
assert _sobjs[
1].DETECTOR is not None, '2nd object has DET=1 so should get decorated'
assert _sobjs[2].DETECTOR is None
# Now try updates!
sobjs1 = specobjs.SpecObjs([sobj1])
sobjs[0]['BOX_WAVE'] = np.arange(2000).astype(float)
sobjs1[0]['DETECTOR'] = tstutils.get_kastb_detector()
header1 = fits.PrimaryHDU().header
sobjs1.write_to_fits(header1, ofile, overwrite=True, update_det=1)
# Test
_sobjs1 = specobjs.SpecObjs.from_fitsfile(ofile)
assert _sobjs1.nobj == 3
assert _sobjs1[2].BOX_WAVE.size == 2000
os.remove(ofile)
def test_all2dobj_write(init_dict):
# Build one
init_dict['detector'] = tstutils.get_kastb_detector()
spec2DObj = spec2dobj.Spec2DObj(**init_dict)
allspec2D = spec2dobj.AllSpec2DObj()
allspec2D['meta']['bkg_redux'] = False
allspec2D['meta']['find_negative'] = False
detname = spec2DObj.detname
allspec2D[detname] = spec2DObj
# Write
ofile = tstutils.data_path('tst_allspec2d.fits')
if os.path.isfile(ofile):
os.remove(ofile)
allspec2D.write_to_fits(ofile)
# Read
_allspec2D = spec2dobj.AllSpec2DObj.from_fits(ofile)
# Check
assert allspec2D.detectors == _allspec2D.detectors, 'Bad read: detector mismatch'
assert allspec2D['meta'] == _allspec2D['meta'], 'Bad read: meta mismatch'
# Try to update it
_allspec2D['meta']['bkg_redux'] = True
_allspec2D[detname].vel_corr = 2.
_allspec2D.write_to_fits(ofile, update_det='DET01')
__allspec2D = spec2dobj.AllSpec2DObj.from_fits(ofile)
assert __allspec2D['meta'] == _allspec2D['meta'], 'Bad read: meta mismatch'
assert __allspec2D['meta'] != allspec2D['meta'], 'Bad read: meta mismatch'
assert __allspec2D[detname].vel_corr == 2., 'Bad update'
os.remove(ofile)
def test_spec2dobj_io(init_dict):
init_dict['detector'] = tstutils.get_kastb_detector()
spec2DObj = spec2dobj.Spec2DObj(**init_dict)
# Write
ofile = tstutils.data_path('tst_spec2d.fits')
if os.path.isfile(ofile):
os.remove(ofile)
spec2DObj.to_file(ofile)
# Read
_spec2DObj = spec2dobj.Spec2DObj.from_file(ofile, spec2DObj.detname)
os.remove(ofile)
def test_all2dobj_hdr(init_dict):
# Build one
init_dict['detector'] = tstutils.get_kastb_detector()
spec2DObj = spec2dobj.Spec2DObj(**init_dict)
allspec2D = spec2dobj.AllSpec2DObj()
allspec2D['meta']['bkg_redux'] = False
allspec2D['meta']['find_negative'] = False
allspec2D[spec2DObj.detname] = spec2DObj
#
kast_file = tstutils.data_path('b1.fits.gz')
header = fits.getheader(kast_file)
spectrograph = load_spectrograph('shane_kast_blue')
# Do it
hdr = allspec2D.build_primary_hdr(header,
spectrograph,
master_dir=tstutils.data_path(''))
# Test it
assert hdr['SKYSUB'] == 'MODEL'
def test_all2dobj_write(init_dict):
# Build one
spec2DObj = spec2dobj.Spec2DObj(**init_dict)
allspec2D = spec2dobj.AllSpec2DObj()
allspec2D['meta']['ir_redux'] = False
allspec2D[1] = spec2DObj
allspec2D[1].detector = tstutils.get_kastb_detector()
# Write
ofile = data_path('tst_allspec2d.fits')
if os.path.isfile(ofile):
os.remove(ofile)
allspec2D.write_to_fits(ofile)
# Read
_allspec2D = spec2dobj.AllSpec2DObj.from_fits(ofile)
# Write again
os.remove(ofile)
_allspec2D.write_to_fits(ofile)
os.remove(ofile)
def test_init(init_dict):
init_dict['detector'] = tstutils.get_kastb_detector()
spec2DObj = spec2dobj.Spec2DObj(**init_dict)
# Check
assert spec2DObj.hdu_prefix == 'DET01-'