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)
