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)