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"
def test_mde_sample_sources(dataset):
sampler = MapDatasetEventSampler(random_state=0)
events = sampler.sample_sources(dataset=dataset)
assert len(events.table["ENERGY_TRUE"]) == 2407
assert_allclose(events.table["ENERGY_TRUE"][0], 2.245024, rtol=1e-5)
assert events.table["ENERGY_TRUE"].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):
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, 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)
def test_mde_sample_sources(dataset):
sampler = MapDatasetEventSampler(random_state=0)
events = sampler.sample_sources(dataset=dataset)
assert len(events.table["ENERGY_TRUE"]) == 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_TRUE"][0], 266.296592804, rtol=1e-5)
assert events.table["RA_TRUE"].unit == "deg"
assert_allclose(events.table["DEC_TRUE"][0], -29.056521954, rtol=1e-5)
assert events.table["DEC_TRUE"].unit == "deg"
assert_allclose(events.table["TIME"][0], 32.73797244764, rtol=1e-5)
assert events.table["TIME"].unit == "s"
assert_allclose(events.table["MC_ID"][0], 1, rtol=1e-5)
