def test_abund_change_file():
"""Can we change the abundance setting: file
This test hard-codes the number of elements expected in the
file.
"""
from sherpa.astro import xspec
elems = {n: i * 0.1 for i, n in enumerate(ELEMENT_NAMES)}
tfh = NamedTemporaryFile(mode='w', suffix='.xspec')
for n in ELEMENT_NAMES:
tfh.write("{}\n".format(elems[n]))
tfh.flush()
oval = xspec.get_xsabund()
try:
xspec.set_xsabund(tfh.name)
abund = xspec.get_xsabund()
out = {n: xspec.get_xsabund(n) for n in ELEMENT_NAMES}
finally:
xspec.set_xsabund(oval)
assert abund == 'file'
for n in ELEMENT_NAMES:
assert out[n] == pytest.approx(elems[n])
def setUp(self):
self._old_logger_level = logger.getEffectiveLevel()
logger.setLevel(logging.ERROR)
from sherpa.astro.io import read_pha
from sherpa.astro import xspec
# Ensure we have a known set of XSPEC settings.
# At present this is just the abundance and cross-section,
# since the cosmology settings do not affect any of the
# models used here.
#
self._xspec_settings = {
'abund': xspec.get_xsabund(),
'xsect': xspec.get_xsxsect()
}
xspec.set_xsabund('angr')
xspec.set_xsxsect('bcmc')
pha_fname = self.make_path("9774.pi")
self.data = read_pha(pha_fname)
self.data.notice(0.5, 7.0)
bkg_fname = self.make_path("9774_bg.pi")
self.bkg = read_pha(bkg_fname)
abs1 = xspec.XSphabs('abs1')
p1 = PowLaw1D('p1')
self.model = abs1 + p1
self.model_mult = abs1 * p1
pi2278 = self.make_path("pi2278.fits")
pi2286 = self.make_path("pi2286.fits")
self.data_pi2278 = read_pha(pi2278)
self.data_pi2286 = read_pha(pi2286)
def test_abund_element():
"""Can we access the elemental settings?
"""
from sherpa.astro import xspec
oval = xspec.get_xsabund()
try:
xspec.set_xsabund('wilm')
h = xspec.get_xsabund('H')
he = xspec.get_xsabund('He')
si = xspec.get_xsabund('Si')
ar = xspec.get_xsabund('Ar')
k = xspec.get_xsabund('K')
fe = xspec.get_xsabund('Fe')
finally:
xspec.set_xsabund(oval)
# These values were found from HEASOFT version 6.19
# spectral/manager/abundances.dat
# The values are given to two decimal places in this file.
# It is not worth testing all settings, since we are not
# testing the XSPEC implementation itself, just our use of it.
#
assert h == pytest.approx(1.0)
assert he == pytest.approx(9.77e-2)
assert si == pytest.approx(1.86e-05)
assert ar == pytest.approx(2.57e-06)
assert k == pytest.approx(0.0)
assert fe == pytest.approx(2.69e-05)
def test_abund_change_file():
"""Can we change the abundance setting: file
This test hard-codes the number of elements expected in the
file.
"""
from sherpa.astro import xspec
elems = {n: i * 0.1 for i, n in enumerate(ELEMENT_NAMES)}
tfh = NamedTemporaryFile(mode='w', suffix='.xspec')
for n in ELEMENT_NAMES:
tfh.write("{}\n".format(elems[n]))
tfh.flush()
oval = xspec.get_xsabund()
try:
xspec.set_xsabund(tfh.name)
abund = xspec.get_xsabund()
out = {n: xspec.get_xsabund(n)
for n in ELEMENT_NAMES}
finally:
xspec.set_xsabund(oval)
assert abund == 'file'
for n in ELEMENT_NAMES:
assert out[n] == pytest.approx(elems[n])
def test_abund_element():
"""Can we access the elemental settings?
"""
from sherpa.astro import xspec
oval = xspec.get_xsabund()
try:
xspec.set_xsabund('wilm')
h = xspec.get_xsabund('H')
he = xspec.get_xsabund('He')
si = xspec.get_xsabund('Si')
ar = xspec.get_xsabund('Ar')
k = xspec.get_xsabund('K')
fe = xspec.get_xsabund('Fe')
finally:
xspec.set_xsabund(oval)
# These values were found from HEASOFT version 6.19
# spectral/manager/abundances.dat
# The values are given to two decimal places in this file.
# It is not worth testing all settings, since we are not
# testing the XSPEC implementation itself, just our use of it.
#
assert h == pytest.approx(1.0)
assert he == pytest.approx(9.77e-2)
assert si == pytest.approx(1.86e-05)
assert ar == pytest.approx(2.57e-06)
assert k == pytest.approx(0.0)
assert fe == pytest.approx(2.69e-05)
def tearDown(self):
from sherpa.astro import xspec
self._xspec_settings = {
'abund': xspec.get_xsabund(),
'xsect': xspec.get_xsxsect()
}
xspec.set_xsabund(self._xspec_settings['abund'])
xspec.set_xsxsect(self._xspec_settings['xsect'])
if hasattr(self, "_old_logger_level"):
logger.setLevel(self._old_logger_level)
def reset_xspec():
from sherpa.astro import xspec
# Ensure we have a known set of XSPEC settings.
# At present this is just the abundance and cross-section,
# since the cosmology settings do not affect any of the
# models used here.
#
abund = xspec.get_xsabund()
xsect = xspec.get_xsxsect()
xspec.set_xsabund('angr')
xspec.set_xsxsect('bcmc')
yield
xspec.set_xsabund(abund)
xspec.set_xsxsect(xsect)
def fix_xspec(clean_astro_ui, hide_logging):
"""As of XSPEC 12.11.1 it's useful to fix abundance/cross sections
rather than rely on reading from the user's directory
This requires XSPEC support.
"""
# As of XSPEC 12.10.1, it is safest to explicitly set
# the state to a known value. For now just pick the
# "easy" settings.
#
abund = xspec.get_xsabund()
xsect = xspec.get_xsxsect()
xspec.set_xsabund('angr')
xspec.set_xsxsect('bcmc')
yield
xspec.set_xsabund(abund)
xspec.set_xsxsect(xsect)
def setUp(self):
self.is_crates_io = False
try:
import sherpa.astro.io
if "sherpa.astro.io.crates_backend" == sherpa.astro.io.backend.__name__:
self.is_crates_io = True
except ImportError:
self.is_crates_io = False
self.old_state = ui._session.__dict__.copy()
self.old_level = logger.getEffectiveLevel()
logger.setLevel(logging.CRITICAL)
# Store XSPEC settings, if applicable
if has_xspec:
self.old_xspec = xspec.get_xsstate()
# As of XSPEC 12.10.1, it is safest to explicitly set
# the state to a known value. For now just pick the
# "easy" settings.
#
xspec.set_xsabund('angr')
xspec.set_xsxsect('bcmc')
