def test_arithmetic(): spec1 = Spectrum.from_powerlaw(1.0, 0.05, 1.0e-4, 0.1, 10.0, 10000) spec2 = Spectrum.from_powerlaw(2.0, 0.01, 1.0e-3, 0.1, 10.0, 10000) spec3 = spec1 + spec2 flux3 = spec1.flux + spec2.flux assert_allclose(spec3.flux, flux3) spec4 = spec3 * 3.0 spec5 = 3.0 * spec3 flux4 = spec3.flux * 3.0 assert_allclose(spec4.flux, spec5.flux) assert_allclose(spec4.flux, flux4) spec6 = spec3 / 2.5 flux6 = spec3.flux / 2.5 assert_allclose(spec6.flux, flux6) spec7 = Spectrum.from_constant(1.0e-4, 0.1, 10.0, 10000) spec8 = spec1 + spec7 assert_allclose(spec8.flux.value, spec1.flux.value + 1.0e-4)
def test_absorption_line(answer_store, answer_dir): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) const_flux = 1.0e-3 line_pos = 1.0 line_width = 0.02 line_amp = 1.0e-5 exp_time = (100.0, "ks") inst_name = "lynx_xgs" spec = Spectrum.from_constant(const_flux, 0.1, 3.0, 100000) spec.add_absorption_line(line_pos, line_width, line_amp) spectrum_answer_testing(spec, "absorption_line_test.h5", answer_store, answer_dir) simulate_spectrum(spec, inst_name, exp_time, "absorption_line_evt.pha", overwrite=True) file_answer_testing("SPECTRUM", "absorption_line_evt.pha", answer_store, answer_dir) os.chdir(curdir) shutil.rmtree(tmpdir)
def test_read_write(): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) spec1 = Spectrum.from_powerlaw(1.0, 0.05, 1.0e-4, 0.1, 10.0, 10000) spec1.write_file("test_spec.dat", overwrite=True) spec2 = Spectrum.from_file("test_spec.dat") assert_allclose(spec1.flux, spec2.flux) assert_allclose(spec1.emid, spec2.emid) assert_allclose(spec1.ebins, spec2.ebins) assert_allclose(spec1.cumspec, spec2.cumspec) os.chdir(curdir) shutil.rmtree(tmpdir)
def test_convolved_spectra(): arf = AuxiliaryResponseFile("xrs_hdxi_3x10.arf") spec1 = Spectrum.from_powerlaw(2.0, 0.01, 1.0, 0.1, 10.0, 1000) cspec1 = ConvolvedSpectrum(spec1, arf) cspec2 = spec1 * arf spec2 = cspec1.deconvolve() assert_array_equal(cspec1.ebins.value, cspec2.ebins.value) assert_array_equal(spec1.ebins.value, spec2.ebins.value) assert_array_equal(cspec1.flux.value, cspec2.flux.value) assert_allclose(spec1.flux.value, spec2.flux.value)
def test_rescale_flux(): spec = Spectrum.from_powerlaw(2.0, 0.01, 1.0, 0.1, 10.0, 10000) spec.rescale_flux(1.0e-4, emin=0.5, emax=7.0, flux_type="photons") f = spec.get_flux_in_band(0.5, 7.0)[0] assert_allclose(1.0e-4, f.value) spec.rescale_flux(1.0e-12, emin=0.4, emax=1.0, flux_type="energy") f = spec.get_flux_in_band(0.4, 1.0)[1] assert_allclose(1.0e-12, f.value)
def to_scaled_spectrum(self, fov, focal_length=None): from xcs_soxs.instrument import FlatResponse fov = parse_value(fov, "arcmin") if focal_length is None: focal_length = self.default_focal_length else: focal_length = parse_value(focal_length, "m") flux = self.flux.value * fov * fov flux *= (focal_length / self.default_focal_length)**2 arf = FlatResponse(self.ebins.value[0], self.ebins.value[-1], 1.0, self.ebins.size - 1) return ConvolvedSpectrum(Spectrum(self.ebins.value, flux), arf)
def plaw_fit(alpha_sim, answer_store, answer_dir): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) nH_sim = 0.02 norm_sim = 1.0e-4 redshift = 0.01 exp_time = (50.0, "ks") area = 40000.0 inst_name = "new_hdxi" spec = Spectrum.from_powerlaw(alpha_sim, redshift, norm_sim, 0.1, 10.0, 20000) spec.apply_foreground_absorption(nH_sim, model="tbabs") spectrum_answer_testing(spec, "power_law_%s.h5" % alpha_sim, answer_store, answer_dir) pt_src_pos = PointSourceModel(30.0, 45.0) sim_cat = SimputCatalog.from_models("plaw_model", "plaw_model", spec, pt_src_pos, exp_time, area, prng=prng) sim_cat.write_catalog(overwrite=True) instrument_simulator("plaw_model_simput.fits", "plaw_model_%s_evt.fits" % alpha_sim, exp_time, inst_name, [30.0, 45.0], instr_bkgnd=False, ptsrc_bkgnd=False, foreground=False, prng=prng) write_spectrum("plaw_model_%s_evt.fits" % alpha_sim, "plaw_model_%s_evt.pha" % alpha_sim, overwrite=True) file_answer_testing("EVENTS", "plaw_model_%s_evt.fits" % alpha_sim, answer_store, answer_dir) file_answer_testing("SPECTRUM", "plaw_model_%s_evt.pha" % alpha_sim, answer_store, answer_dir) os.chdir(curdir) shutil.rmtree(tmpdir)
def test_append(): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) exp_time = (50.0, "ks") area = (4.0, "m**2") ra0 = 30.0 dec0 = 45.0 spec = Spectrum.from_powerlaw(1.1, 0.05, 1.0e-4, 0.1, 10.0, 10000) e1 = spec.generate_energies(exp_time, area, prng=prng) ra1, dec1 = PointSourceModel(ra0 + 0.05, dec0 + 0.05).generate_coords(e1.size, prng=prng) e2 = spec.generate_energies(exp_time, area, prng=prng) ra2, dec2 = PointSourceModel(ra0 - 0.05, dec0 - 0.05).generate_coords(e1.size, prng=prng) write_photon_list("pt_src", "pt_src1", e1.flux, ra1, dec1, e1, overwrite=True) write_photon_list("pt_src", "pt_src2", e2.flux, ra2, dec2, e2, append=True) assert os.path.exists("pt_src_simput.fits") assert os.path.exists("pt_src1_phlist.fits") assert os.path.exists("pt_src2_phlist.fits") f = pyfits.open("pt_src_simput.fits") cat = f["SRC_CAT"].data["SPECTRUM"] assert cat[0] == "pt_src1_phlist.fits[PHLIST,1]" assert cat[1] == "pt_src2_phlist.fits[PHLIST,1]" f.close() os.chdir(curdir) shutil.rmtree(tmpdir)
def test_emission_line(answer_store, answer_dir): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) const_flux = 1.0e-4 line_pos = 5.0 line_width = 0.02 line_amp = 1.0e-5 exp_time = (100.0, "ks") area = 30000.0 inst_name = "lynx_lxm" spec = Spectrum.from_constant(const_flux, 1.0, 10.0, 20000) spec.add_emission_line(line_pos, line_width, line_amp) spectrum_answer_testing(spec, "emission_line_test.h5", answer_store, answer_dir) pt_src_pos = PointSourceModel(30.0, 45.0) sim_cat = SimputCatalog.from_models("emission_line", "emission_line", spec, pt_src_pos, exp_time, area, prng=prng) sim_cat.write_catalog(overwrite=True) instrument_simulator("emission_line_simput.fits", "emission_line_evt.fits", exp_time, inst_name, [30.0, 45.0], instr_bkgnd=False, ptsrc_bkgnd=False, foreground=False, prng=prng) write_spectrum("emission_line_evt.fits", "emission_line_evt.pha", overwrite=True) file_answer_testing("EVENTS", "emission_line_evt.fits", answer_store, answer_dir) file_answer_testing("SPECTRUM", "emission_line_evt.pha", answer_store, answer_dir) os.chdir(curdir) shutil.rmtree(tmpdir)
def to_spectrum(self, fov): fov = parse_value(fov, "arcmin") flux = self.flux.value * fov * fov return Spectrum(self.ebins.value, flux)