def test_mde_sample_edisp(dataset): sampler = MapDatasetEventSampler(random_state=0) events = sampler.sample_sources(dataset=dataset) events = sampler.sample_edisp(dataset.edisp, events) assert len(events.table) == 348 assert_allclose(events.table["ENERGY"][0], 3.53609085283, rtol=1e-5) assert events.table["ENERGY"].unit == "TeV" assert_allclose(events.table["RA_TRUE"][0], 266.296592804, rtol=1e-5) assert events.table["RA_TRUE"].unit == "deg" assert_allclose(events.table["DEC_TRUE"][0], -29.05652195, rtol=1e-5) assert events.table["DEC_TRUE"].unit == "deg" assert_allclose(events.table["MC_ID"][0], 1, rtol=1e-5)
def test_mde_sample_edisp(dataset): sampler = MapDatasetEventSampler(random_state=0) events = sampler.sample_sources(dataset=dataset) events = sampler.sample_edisp(dataset.edisp, events) assert len(events.table) == 2407 assert_allclose(events.table["ENERGY"][0], 2.24502, rtol=1e-5) assert events.table["ENERGY"].unit == "TeV" assert_allclose(events.table["RA_TRUE"][0], 266.912888, rtol=1e-5) assert events.table["RA_TRUE"].unit == "deg" assert_allclose(events.table["DEC_TRUE"][0], -29.034641, rtol=1e-5) assert events.table["DEC_TRUE"].unit == "deg" assert_allclose(events.table["MC_ID"][0], 1, rtol=1e-5)
def test_mde_sample_edisp(dataset, models): dataset.models = models sampler = MapDatasetEventSampler(random_state=0) events = sampler.sample_sources(dataset=dataset) events = sampler.sample_edisp(dataset.edisp, events) assert len(events.table) == 88 assert_allclose(events.table["ENERGY"][0], 2.751205, rtol=1e-5) assert events.table["ENERGY"].unit == "TeV" assert_allclose(events.table["RA_TRUE"][0], 266.559566, rtol=1e-5) assert events.table["RA_TRUE"].unit == "deg" assert_allclose(events.table["DEC_TRUE"][0], -28.742429, rtol=1e-5) assert events.table["DEC_TRUE"].unit == "deg" assert_allclose(events.table["MC_ID"][0], 1, rtol=1e-5)