def test_issue_16(): """Check one of the examples from #16""" s = Session() s._add_model_types(sherpa.models.basic) s.dataspace1d(0.01, 11, 0.01, id=1) s.dataspace1d(2, 5, 0.1, id="tst") s.set_source(1, 'powlaw1d.pl1') s.set_source('tst', 'powlaw1d.pltst') assert s.list_data_ids() == [1, 'tst'] assert s.list_model_ids() == [1, 'tst'] assert s.list_model_components() == ['pl1', 'pltst'] s.delete_model(id='tst') s.delete_model_component("pltst") s.delete_data(id='tst') assert s.list_data_ids() == [1] assert s.list_model_ids() == [1] assert s.list_model_components() == ['pl1'] s.delete_model(id=1) s.delete_model_component("pl1") s.delete_data(id=1) assert s.list_data_ids() == [] assert s.list_model_ids() == [] assert s.list_model_components() == []
def test_save_restore(tmpdir): outfile = tmpdir.join("sherpa.save") session = Session() session.load_arrays(1, TEST, TEST2) session.save(str(outfile), clobber=True) session.clean() assert set() == set(session.list_data_ids()) session.restore(str(outfile)) assert {1, } == set(session.list_data_ids()) assert_array_equal(TEST, session.get_data(1).get_indep()[0]) assert_array_equal(TEST2, session.get_data(1).get_dep())
def test_save_restore(tmpdir): outfile = tmpdir.join("sherpa.save") session = Session() session.load_arrays(1, TEST, TEST2) session.save(str(outfile), clobber=True) session.clean() assert set() == set(session.list_data_ids()) session.restore(str(outfile)) assert {1, } == set(session.list_data_ids()) assert_array_equal(TEST, session.get_data(1).get_indep()[0]) assert_array_equal(TEST2, session.get_data(1).get_dep())
def test_list_ids(): session = Session() session.load_arrays(1, TEST, TEST) session.load_arrays("1", TEST, TEST2) # order of 1 and "1" is not determined assert {1, "1"} == set(session.list_data_ids()) assert_array_equal(TEST2, session.get_data('1').get_dep()) assert_array_equal(TEST, session.get_data(1).get_dep())
def test_list_ids(): session = Session() session.load_arrays(1, [1, 2, 3], [1, 2, 3]) session.load_arrays("1", [1, 2, 3], [4, 5, 6]) # order of 1 and "1" is not determined assert {1, "1"} == set(session.list_data_ids()) assert_array_equal([4, 5, 6], session.get_data('1').get_dep()) assert_array_equal([1, 2, 3], session.get_data(1).get_dep())
def test_list_ids(): session = Session() session.load_arrays(1, TEST, TEST) session.load_arrays("1", TEST, TEST2) # order of 1 and "1" is not determined assert {1, "1"} == set(session.list_data_ids()) assert_array_equal(TEST2, session.get_data('1').get_dep()) assert_array_equal(TEST, session.get_data(1).get_dep())
from sherpa.ui.utils import Session import sherpa.models.basic x = [100, 200, 300, 400] y = [10, 12, 9, 13] s = Session() s._add_model_types(sherpa.models.basic) s.load_arrays(1, x, y) print("# list_data_ids") print(s.list_data_ids()) print("# get_data()") print(repr(s.get_data())) print("# get_data()") print(s.get_data()) print("# get_stat_name/get_method_name") print(s.get_stat_name()) print(s.get_method_name()) s.set_stat('cash') s.set_method('simplex') s.set_source('const1d.mdl') print("# mdl") print(mdl)