def test_custom_prefix():
container = HillasParametersContainer(
x=1 * u.m, y=1 * u.m, length=1 * u.m, width=1 * u.m
)
container.prefix = "custom"
with tempfile.NamedTemporaryFile() as f:
with HDF5TableWriter(f.name, group_name="dl1", add_prefix=True) as writer:
writer.write("params", container)
with HDF5TableReader(f.name) as reader:
generator = reader.read(
"/dl1/params", HillasParametersContainer(), prefixes="custom"
)
read_container = next(generator)
assert isinstance(read_container, HillasParametersContainer)
for value, read_value in zip(
container.as_dict().values(), read_container.as_dict().values()
):
np.testing.assert_equal(value, read_value)
def test_custom_prefix(tmp_path):
path = tmp_path / "test.h5"
container = HillasParametersContainer(fov_lon=1 * u.deg,
fov_lat=1 * u.deg,
length=1 * u.deg,
width=1 * u.deg)
container.prefix = "custom"
with HDF5TableWriter(path, group_name="dl1", add_prefix=True) as writer:
writer.write("params", container)
with HDF5TableReader(path) as reader:
generator = reader.read("/dl1/params",
HillasParametersContainer(),
prefixes="custom")
read_container = next(generator)
assert isinstance(read_container, HillasParametersContainer)
for value, read_value in zip(container.as_dict().values(),
read_container.as_dict().values()):
np.testing.assert_equal(value, read_value)
