def test_beta_model_flux(answer_store, answer_dir): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) r_c = 20.0 beta = 1.0 prng = 34 beta_src_pos = BetaModel(ra0, dec0, r_c, beta) sim_cat = SimputCatalog.from_models("beta", "beta", spec, beta_src_pos, exp_time, area, prng=prng) sim_cat.write_catalog(overwrite=True) instrument_simulator("beta_simput.fits", "beta_flux_evt.fits", exp_time, "acisi_cy0", [ra0, dec0], ptsrc_bkgnd=False, instr_bkgnd=False, foreground=False, roll_angle=37.0, prng=prng) wspec = spec.new_spec_from_band(0.5, 7.0) make_exposure_map("beta_flux_evt.fits", "beta_expmap.fits", wspec.emid.value, weights=wspec.flux.value, overwrite=True) write_radial_profile("beta_flux_evt.fits", "beta_flux_evt_profile.fits", [ra0, dec0], 0.0, 100.0, 200, ctr_type="celestial", emin=0.5, emax=7.0, expmap_file="beta_expmap.fits", overwrite=True) file_answer_testing("EVENTS", "beta_flux_evt.fits", answer_store, answer_dir) file_answer_testing("PROFILE", "beta_flux_evt_profile.fits", answer_store, answer_dir) os.chdir(curdir) shutil.rmtree(tmpdir)
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_beta_model(answer_store, answer_dir): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) prng = 32 r_c = 20.0 beta = 1.0 exp_time = Quantity(500.0, "ks") beta_src_pos = BetaModel(ra0, dec0, r_c, beta) sim_cat = SimputCatalog.from_models("beta", "beta", spec, beta_src_pos, exp_time, area, prng=prng) sim_cat.write_catalog(overwrite=True) instrument_simulator("beta_simput.fits", "beta_evt.fits", exp_time, "acisi_cy0", [ra0, dec0], ptsrc_bkgnd=False, instr_bkgnd=False, foreground=False, prng=prng) write_radial_profile("beta_evt.fits", "beta_evt_profile.fits", [ra0, dec0], 0.0, 100.0, 200, ctr_type="celestial", emin=0.5, emax=7.0, overwrite=True) file_answer_testing("EVENTS", "beta_evt.fits", answer_store, answer_dir) file_answer_testing("PROFILE", "beta_evt_profile.fits", answer_store, answer_dir) os.chdir(curdir) shutil.rmtree(tmpdir)
def test_point_source(): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) pt_src_pos = PointSourceModel(ra0, dec0) sim_cat = SimputCatalog.from_models("pt_src", "pt_src", spec, pt_src_pos, exp_time, area, prng=prng) sim_cat.write_catalog(overwrite=True) inst = get_instrument_from_registry("hdxi") inst["name"] = "hdxi_big_psf" inst["psf"] = ["gaussian", 5.0] add_instrument_to_registry(inst) instrument_simulator("pt_src_simput.fits", "pt_src_evt.fits", exp_time, "hdxi_big_psf", [ra0, dec0], ptsrc_bkgnd=False, instr_bkgnd=False, foreground=False, prng=prng) psf_scale = inst["psf"][1] dtheta = inst["fov"] * 60.0 / inst["num_pixels"] f = pyfits.open("pt_src_evt.fits") x = f["EVENTS"].data["X"] y = f["EVENTS"].data["Y"] f.close() scalex = np.std(x) * sigma_to_fwhm * dtheta scaley = np.std(y) * sigma_to_fwhm * dtheta assert (scalex - psf_scale) / psf_scale < 0.03 assert (scaley - psf_scale) / psf_scale < 0.03 os.chdir(curdir) shutil.rmtree(tmpdir)
def test_annulus(answer_store, answer_dir): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) r_in = 10.0 r_out = 30.0 ann_pos = AnnulusModel(ra0, dec0, r_in, r_out) sim_cat = SimputCatalog.from_models("ann", "ann", spec, ann_pos, exp_time, area, prng=prng) sim_cat.write_catalog(overwrite=True) instrument_simulator("ann_simput.fits", "ann_evt.fits", exp_time, "hdxi", [ra0, dec0], ptsrc_bkgnd=False, instr_bkgnd=False, foreground=False, prng=prng) write_radial_profile("ann_evt.fits", "ann_evt_profile.fits", [ra0, dec0], 1.1 * r_in, 0.9 * r_out, 100, ctr_type="celestial", emin=0.5, emax=7.0, overwrite=True) file_answer_testing("EVENTS", "ann_evt.fits", answer_store, answer_dir) file_answer_testing("PROFILE", "ann_evt_profile.fits", answer_store, answer_dir) 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 test_add_background(): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) prng1 = RandomState(29) prng2 = RandomState(29) ra0 = 30.0 dec0 = 45.0 ra1 = 22.0 dec1 = 22.0 exp_time = 50000.0 ra = np.array([]) dec = np.array([]) e = np.array([]) empty_cat = { "ra": [ra], "dec": [dec], "energy": [e], "flux": [0.0], "emin": [0.1], "emax": [10.0], "sources": ["empty"] } instrument_simulator(empty_cat, "evt1.fits", exp_time, "hdxi", [ra0, dec0], prng=prng1, overwrite=True) make_background_file("bkg_evt.fits", exp_time, "hdxi", [ra0, dec0], prng=prng2, overwrite=True) instrument_simulator(empty_cat, "evt2.fits", exp_time, "hdxi", [ra1, dec1], bkgnd_file="bkg_evt.fits", prng=prng2, overwrite=True) f1 = pyfits.open("evt1.fits") f2 = pyfits.open("evt2.fits") for key in ["X", "Y", "ENERGY", "PHA"]: assert_allclose( f1["EVENTS"].data[key], f2["EVENTS"].data[key], ) f1.close() f2.close() os.chdir(curdir) shutil.rmtree(tmpdir)