def test_sensitivity(mock_show):
hiltner_spectrum1D = Spectrum1D(log_file_name=None)
sens = FluxCalibration(log_file_name=None)
# Standard count
count = np.loadtxt(
os.path.join(HERE, "test_data", "test_full_run_standard_count.csv"),
delimiter=",",
skiprows=1,
)[:, 0]
wavelength = np.loadtxt(
os.path.join(
HERE, "test_data", "test_full_run_standard_wavelength.csv"
),
skiprows=1,
)
hiltner_spectrum1D.add_count(count)
hiltner_spectrum1D.add_wavelength(wavelength)
sens.from_spectrum1D(hiltner_spectrum1D)
# Load standard star from literature
sens.load_standard("hiltner102")
sens.inspect_standard(
display=False,
return_jsonstring=True,
save_fig=True,
fig_type="iframe+png",
filename=os.path.join(HERE, "test_output", "fluxcal_inspect_standard"),
)
sens.inspect_standard(display=True)
sens.get_sensitivity()
# Get back the spectrum1D and merge
hiltner_spectrum1D.merge(sens.get_spectrum1D())
# Save a FITS file
sens.save_fits(
output="sensitivity",
filename=os.path.join(HERE, "test_output", "test_sensitivity"),
overwrite=True,
)
# Save a CSV file
sens.save_csv(
output="sensitivity",
filename=os.path.join(HERE, "test_output", "test_sensitivity"),
overwrite=True,
)
def test_fluxcalibration(mock_show):
hiltner_spectrum1D = Spectrum1D(log_file_name=None)
lhs6328_spectrum1D = Spectrum1D(log_file_name=None)
fluxcalibrator = FluxCalibration(log_file_name=None)
# Science and Standard counts
standard_count = np.loadtxt(
os.path.join(HERE, "test_data", "test_full_run_standard_count.csv"),
delimiter=",",
skiprows=1,
)[:, 0]
science_count = np.loadtxt(
os.path.join(HERE, "test_data", "test_full_run_science_0_count.csv"),
delimiter=",",
skiprows=1,
)[:, 0]
wavelength = np.loadtxt(
os.path.join(
HERE, "test_data", "test_full_run_standard_wavelength.csv"
),
skiprows=1,
)
hiltner_spectrum1D.add_count(standard_count)
hiltner_spectrum1D.add_wavelength(wavelength)
lhs6328_spectrum1D.add_count(science_count)
lhs6328_spectrum1D.add_wavelength(wavelength)
# Add the standard spectrum1D to the flux calibrator
fluxcalibrator.from_spectrum1D(hiltner_spectrum1D)
# Load standard star from literature
fluxcalibrator.load_standard("hiltner102")
fluxcalibrator.get_sensitivity()
# Get back the spectrum1D and merge
fluxcalibrator.apply_flux_calibration(
lhs6328_spectrum1D,
inspect=True,
display=False,
return_jsonstring=True,
save_fig=True,
fig_type="iframe+png",
filename=os.path.join(HERE, "test_output", "fluxcal_flux_calibration"),
)
fluxcalibrator.apply_flux_calibration(lhs6328_spectrum1D, display=True)