def test_delete_bkg_model(clean_astro_ui):
"""Check we can delete a background model"""
channels = np.arange(1, 5)
counts = np.zeros(channels.size)
d = ui.DataPHA('src', channels, counts)
b = ui.DataPHA('bkg', channels, counts)
d.set_background(b, id=2)
ui.set_data(d)
ui.set_bkg_source(ui.const1d.bmdl + ui.gauss1d.gmdl, bkg_id=2)
assert ui.get_bkg_source(bkg_id=2) is not None
ui.delete_bkg_model(bkg_id=2)
# Expression has been removed
#
with pytest.raises(ModelErr) as exc:
ui.get_bkg_source(bkg_id=2)
assert str(
exc.value) == 'background model 2 for data set 1 has not been set'
# components still exist
mdls = ui.list_model_components()
assert set(mdls) == set(['bmdl', 'gmdl'])
def test_fix_background_id_error_checks2():
"""Check error handling of background id"""
ui.load_arrays(2, [1, 2, 3], [5, 4, 3], ui.DataPHA)
bkg = ui.DataPHA('bkg', np.asarray([1, 2, 3]), [1, 1, 0])
ui.set_bkg(2, bkg)
with pytest.raises(IdentifierErr) as exc:
ui.get_bkg_source(id=2, bkg_id='bkg')
assert str(exc.value) == "identifier 'bkg' is a reserved word"
def test_fix_background_id_error_checks1():
"""Check error handling of background id"""
ui.load_arrays(2, [1, 2, 3], [5, 4, 3], ui.DataPHA)
bkg = ui.DataPHA('bkg', np.asarray([1, 2, 3]), [1, 1, 0])
ui.set_bkg(2, bkg)
with pytest.raises(ArgumentTypeErr) as exc:
ui.get_bkg_source(id=2, bkg_id=bkg)
assert str(exc.value) == 'identifiers must be integers or strings'
def test_bkg_id_get_bkg_source(clean_astro_ui):
"""Check the error message when the background model has not been set (issue #943)"""
ui.set_default_id('x')
ui.load_arrays('x', [1, 2, 3], [5, 4, 3], ui.DataPHA)
bkg = ui.DataPHA('bkg', np.asarray([1, 2, 3]), [1, 1, 0])
ui.set_bkg(bkg)
with pytest.raises(ModelErr) as exc:
ui.get_bkg_source()
assert str(
exc.value) == 'background model 1 for data set x has not been set'
def test_delete_bkg_model_with_bkgid(id, clean_astro_ui):
"""Check we call delete_bkg_model with non-default bkg_id"""
ui.load_arrays(id, [1, 2, 3], [5, 4, 3], ui.DataPHA)
bkg = ui.DataPHA('bkg', np.asarray([1, 2, 3]), [1, 1, 0])
ui.set_bkg(id, bkg, bkg_id=2)
ui.set_bkg_source(id, ui.const1d.bmdl, bkg_id=2)
assert ui.list_model_components() == ['bmdl']
assert ui.get_bkg_source(id, 2).name == 'const1d.bmdl'
ui.delete_bkg_model(id, bkg_id=2)
assert ui.list_model_components() == ['bmdl']
with pytest.raises(ModelErr) as exc:
ui.get_bkg_source(id, 2)
emsg = 'background model 2 for data set {} has not been set'.format(id)
assert str(exc.value) == emsg