def test_mde_sample_psf(dataset):
sampler = MapDatasetEventSampler(random_state=0)
events = sampler.sample_sources(dataset=dataset)
events = sampler.sample_psf(dataset.psf, events)
assert len(events.table) == 348
assert_allclose(events.table["ENERGY_TRUE"][0], 3.536090852832, rtol=1e-5)
assert events.table["ENERGY_TRUE"].unit == "TeV"
assert_allclose(events.table["RA"][0], 266.2757792220, rtol=1e-5)
assert events.table["RA"].unit == "deg"
assert_allclose(events.table["DEC"][0], -29.030939, rtol=1e-5)
assert events.table["DEC"].unit == "deg"
def test_mde_sample_psf(dataset):
sampler = MapDatasetEventSampler(random_state=0)
events = sampler.sample_sources(dataset=dataset)
events = sampler.sample_psf(dataset.psf, events)
assert len(events.table) == 2407
assert_allclose(events.table["ENERGY_TRUE"][0], 2.2450239, rtol=1e-5)
assert events.table["ENERGY_TRUE"].unit == "TeV"
assert_allclose(events.table["RA"][0], 266.895383, rtol=1e-5)
assert events.table["RA"].unit == "deg"
assert_allclose(events.table["DEC"][0], -29.052050, rtol=1e-5)
assert events.table["DEC"].unit == "deg"
def test_mde_sample_psf(dataset, models):
dataset.models = models
sampler = MapDatasetEventSampler(random_state=0)
events = sampler.sample_sources(dataset=dataset)
events = sampler.sample_psf(dataset.psf, events)
assert len(events.table) == 88
assert_allclose(events.table["ENERGY_TRUE"][0], 2.751205, rtol=1e-5)
assert events.table["ENERGY_TRUE"].unit == "TeV"
assert_allclose(events.table["RA"][0], 266.556053, rtol=1e-5)
assert events.table["RA"].unit == "deg"
assert_allclose(events.table["DEC"][0], -28.746459, rtol=1e-5)
assert events.table["DEC"].unit == "deg"
