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)