def test_source_methods_with_full_model(self): from sherpa.utils.err import IdentifierErr ui.load_data('full', self.ascii) ui.set_full_model('full', 'powlaw1d.p1') # Test Case 1 try: ui.get_source('full') except IdentifierErr as e: self.assertRegex( str(e), "Convolved model\n.*\n is set for dataset full. You should use get_model instead.", str(e)) try: ui.plot_source('full') except IdentifierErr as e: self.assertRegex( str(e), "Convolved model\n.*\n is set for dataset full. You should use plot_model instead.", str(e)) # Test Case 2 ui.set_source('full', 'powlaw1d.p2') ui.get_source('full') # Test Case 3 ui.load_data('not_full', self.ascii) try: ui.get_source('not_full') except IdentifierErr as e: self.assertEqual( 'source not_full has not been set, consider using set_source() or set_model()', str(e))
def test_ui_source_methods_with_full_model(clean_ui, setup_ui_full): ui.load_data('full', setup_ui_full.ascii) ui.set_full_model('full', 'powlaw1d.p1') # Test Case 1 with pytest.raises(IdentifierErr) as exc: ui.get_source('full') emsg = "Convolved model\n'powlaw1d.p1'\n is set for dataset full. You should use get_model instead." assert str(exc.value) == emsg with pytest.raises(IdentifierErr) as exc: ui.plot_source('full') emsg = "Convolved model\n'powlaw1d.p1'\n is set for dataset full. You should use plot_model instead." assert str(exc.value) == emsg with pytest.raises(IdentifierErr) as exc: ui.get_source_plot('full') emsg = "Convolved model\n'powlaw1d.p1'\n is set for dataset full. You should use get_model_plot instead." assert str(exc.value) == emsg # Test Case 2 ui.set_source('full', 'powlaw1d.p2') ui.get_source('full') # Test Case 3 ui.load_data('not_full', setup_ui_full.ascii) with pytest.raises(IdentifierErr) as exc: ui.get_source('not_full') emsg = 'source not_full has not been set, consider using set_source() or set_model()' assert emsg == str(exc.value)
def test_source_methods_with_full_model(self): from sherpa.utils.err import IdentifierErr ui.load_data('full', self.ascii) ui.set_full_model('full', 'powlaw1d.p1') # Test Case 1 try: ui.get_source('full') except IdentifierErr as e: self.assertRegexpMatches(str(e), "Convolved model\n.*\n is set for dataset full. You should use get_model instead.", str(e)) try: ui.plot_source('full') except IdentifierErr as e: self.assertEquals("Convolved model\n'p1'\n is set for dataset full. You should use plot_model instead.", str(e)) # Test Case 2 ui.set_source('full', 'powlaw1d.p2') ui.get_source('full') # Test Case 3 ui.load_data('not_full', self.ascii) try: ui.get_source('not_full') except IdentifierErr as e: self.assertEquals('source not_full has not been set, consider using set_source() or set_model()', str(e))