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,
prng=prng)
file_answer_testing("SPECTRUM", "absorption_line_evt.pha", answer_store,
answer_dir)
os.chdir(curdir)
shutil.rmtree(tmpdir)
def test_arithmetic():
spec1 = Spectrum.from_powerlaw(1.0, 0.05, 1.0e-4)
spec2 = Spectrum.from_powerlaw(2.0, 0.01, 1.0e-3)
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)
spec8 = spec1+spec7
assert_allclose(spec8.flux.value, spec1.flux.value+1.0e-4)
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_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)