def hdu_index_table():
table = HDUIndexTable(rows=[{
"OBS_ID": 42,
"HDU_TYPE": "events",
"HDU_CLASS": "spam42",
"FILE_DIR": "a",
"FILE_NAME": "b",
"HDU_NAME": "c",
}])
table.meta["BASE_DIR"] = "spam"
return table
def test_hdu_index_table_hd_hap():
"""Test HESS HAP-HD data access."""
hdu_index = HDUIndexTable.read(
"$GAMMAPY_DATA/hess-dl3-dr1/hdu-index.fits.gz")
assert "BASE_DIR" in hdu_index.meta
assert hdu_index.base_dir == make_path("$GAMMAPY_DATA/hess-dl3-dr1")
# A few valid queries
location = hdu_index.hdu_location(obs_id=23523, hdu_type="events")
hdu = location.get_hdu()
assert hdu.name == "EVENTS"
# The next line is to check if the HDU is still accessible
# See https://github.com/gammapy/gammapy/issues/1775
assert hdu.filebytes() == 224640
assert location.path(
abs_path=False) == Path("data/hess_dl3_dr1_obs_id_023523.fits.gz")
path1 = str(location.path(abs_path=True))
path2 = str(location.path(abs_path=False))
assert path1.endswith(path2)
location = hdu_index.hdu_location(obs_id=23523, hdu_class="psf_table")
assert location.path(
abs_path=False) == Path("data/hess_dl3_dr1_obs_id_023523.fits.gz")
location = hdu_index.hdu_location(obs_id=23523, hdu_type="psf")
assert location.path(
abs_path=False) == Path("data/hess_dl3_dr1_obs_id_023523.fits.gz")
# A few invalid queries
with pytest.raises(IndexError) as excinfo:
hdu_index.hdu_location(obs_id=42, hdu_class="psf_3gauss")
msg = "No entry available with OBS_ID = 42"
assert str(excinfo.value) == msg
with pytest.raises(IndexError) as excinfo:
hdu_index.hdu_location(obs_id=23523, hdu_type="bkg")
msg = "No HDU found matching: OBS_ID = 23523, HDU_TYPE = bkg, HDU_CLASS = None"
assert str(excinfo.value) == msg
with pytest.raises(ValueError) as excinfo:
hdu_index.hdu_location(obs_id=23523)
msg = "You have to specify `hdu_type` or `hdu_class`."
assert str(excinfo.value) == msg
with pytest.raises(ValueError) as excinfo:
hdu_index.hdu_location(obs_id=23523, hdu_type="invalid")
msg = "Invalid hdu_type: invalid. Valid values are: ['events', 'gti', 'aeff', 'edisp', 'psf', 'bkg']"
assert str(excinfo.value) == msg
with pytest.raises(ValueError) as excinfo:
hdu_index.hdu_location(obs_id=23523, hdu_class="invalid")
msg = "Invalid hdu_class: invalid. Valid values are: ['events', 'gti', 'aeff_2d', 'edisp_2d', 'psf_table', 'psf_3gauss', 'psf_king', 'bkg_2d', 'bkg_3d']"
assert str(excinfo.value) == msg
def test_hdu_index():
hdu_index = HDUIndexTable.read(
'$GAMMAPY_EXTRA/datasets/hess-crab4-hd-hap-prod2/hdu-index.fits.gz')
hdu_index.info()
location = hdu_index.hdu_location(obs_id=23523, hdu_type='events')
location.info()
def test_hdu_index():
hdu_index = HDUIndexTable.read('$GAMMAPY_EXTRA/datasets/hess-crab4-hd-hap-prod2/hdu-index.fits.gz')
hdu_index.info()
location = hdu_index.hdu_location(obs_id=23523, hdu_type='events')
location.info()