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.parse_args( [coadd_ifile, '--test_spec_path', data_path('')])) hdu = io.fits_open(coadd_ofile) assert hdu[1].header['EXT_MODE'] == 'OPT' assert hdu[1].header['FLUXED'] is False # Clean up hdu.close() 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.parse_args( [coadd_ifile, '--test_spec_path', data_path('')])) hdu = io.fits_open(coadd_ofile) assert hdu[1].header['EXT_MODE'] == 'OPT' assert hdu[1].header['FLUXED'] is False # Clean up hdu.close() os.remove(parfile) os.remove(coadd_ofile)
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_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_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())