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)