def test_coadd2():
""" Test using a list of object names
"""
coadd_file = data_path('coadd_UGC3672A_red_objlist.yaml')
args = coadd_1dspec.parser([coadd_file])
# Main
gparam, ex_value, flux_value, iobj, outfile, files, obj_kwargs, = coadd_1dspec.main(
args, unit_test=True, path=data_path('./'))
# Test
assert len(iobj) == len(files)
# Crash it
coadd_file = data_path('coadd_UGC3672A_red_badlist.yaml')
args = coadd_1dspec.parser([coadd_file])
with pytest.raises(IOError):
gparam, ex_value, flux_value, iobj, outfile, files, _ = coadd_1dspec.main(
args, unit_test=True, path=data_path('./'))
def test_coadd():
coadd_file = data_path('coadd_UGC3672A_red.yaml')
args = coadd_1dspec.parser([coadd_file])
# Main
gparam, ex_value, flux_value, iobj, outfile, files, local_kwargs = coadd_1dspec.main(
args, unit_test=True, path=data_path('./'))
# Test
assert len(gparam) == 2
assert isinstance(gparam, dict)
assert ex_value == 'opt'
assert flux_value is True
assert iobj == 'O210-S1467-D02-I0012'
assert outfile == 'UGC3672A_r.fits'
assert len(files) == 4
assert isinstance(local_kwargs, dict)
assert 'otol' in list(local_kwargs.keys())
assert 'scale_method' in list(gparam.keys())
def test_coadd1d_2():
"""
Test combining Echelle
"""
# NOTE: flux_value is False
parfile = 'coadd1d.par'
if os.path.isfile(parfile):
os.remove(parfile)
coadd_ofile = data_path('pisco_coadd.fits')
if os.path.isfile(coadd_ofile):
os.remove(coadd_ofile)
coadd_ifile = data_path('gemini_gnirs_32_sb_sxd.coadd1d')
coadd_1dspec.main(coadd_1dspec.parser([coadd_ifile, '--test_spec_path', data_path('')]))
hdu = fits.open(coadd_ofile)
assert hdu[0].header['NSPEC'] == 6, 'Bad number of spectra'
assert [h.name for h in hdu] == ['PRIMARY', 'OBJ0001-SPEC0001-OPT'], 'Bad extensions'
assert np.all([c.split('_')[0] == 'OPT' for c in hdu[1].columns.names]), 'Bad columns'
# Clean up
os.remove(parfile)
os.remove(coadd_ofile)
def test_coadd1d_1():
"""
Test basic coadd using Shane Kast blue
"""
# NOTE: flux_value is False
parfile = 'coadd1d.par'
if os.path.isfile(parfile):
os.remove(parfile)
coadd_ofile = data_path('J1217p3905_coadd.fits')
if os.path.isfile(coadd_ofile):
os.remove(coadd_ofile)
coadd_ifile = data_path('shane_kast_blue.coadd1d')
coadd_1dspec.main(coadd_1dspec.parser([coadd_ifile, '--test_spec_path', data_path('')]))
hdu = fits.open(coadd_ofile)
assert hdu[0].header['NSPEC'] == 1, 'Bad number of spectra'
assert [h.name for h in hdu] == ['PRIMARY', 'OBJ0001-SPEC0001-OPT'], 'Bad extensions'
assert np.all([c.split('_')[0] == 'OPT' for c in hdu[1].columns.names]), 'Bad columns'
# Clean up
os.remove(parfile)
os.remove(coadd_ofile)
def test_coadd1d_2():
"""
Test combining Echelle
"""
# NOTE: flux_value is False
parfile = 'coadd1d.par'
if os.path.isfile(parfile):
os.remove(parfile)
coadd_ofile = data_path('pisco_coadd.fits')
if os.path.isfile(coadd_ofile):
os.remove(coadd_ofile)
coadd_ifile = data_path('gemini_gnirs_32_sb_sxd.coadd1d')
coadd_1dspec.main(
coadd_1dspec.parser([coadd_ifile, '--test_spec_path',
data_path('')]))
hdu = fits.open(coadd_ofile)
assert hdu[1].header['EXT_MODE'] == 'OPT'
assert hdu[1].header['FLUXED'] is False
# Clean up
os.remove(parfile)
os.remove(coadd_ofile)
def test_coadd1d_1():
"""
Test basic coadd using shane_kast_blue
"""
# NOTE: flux_value is False
parfile = 'coadd1d.par'
if os.path.isfile(parfile):
os.remove(parfile)
coadd_ofile = data_path('J1217p3905_coadd.fits')
if os.path.isfile(coadd_ofile):
os.remove(coadd_ofile)
coadd_ifile = data_path('shane_kast_blue.coadd1d')
coadd_1dspec.main(
coadd_1dspec.parser([coadd_ifile, '--test_spec_path',
data_path('')]))
hdu = fits.open(coadd_ofile)
assert hdu[1].header['EXT_MODE'] == 'OPT'
assert hdu[1].header['FLUXED'] is False
# Clean up
os.remove(parfile)
os.remove(coadd_ofile)