def test_run_get_virtual_dataset(mock_fxe_raw_run): run = RunDirectory(mock_fxe_raw_run) ds = run.get_virtual_dataset('FXE_DET_LPD1M-1/DET/6CH0:xtdf', 'image.data') assert isinstance(ds, h5py.Dataset) assert ds.is_virtual assert ds.shape == (61440, 1, 256, 256) # Across two sequence files ds = run.get_virtual_dataset('FXE_XAD_GEC/CAM/CAMERA:daqOutput', 'data.image.pixels') assert isinstance(ds, h5py.Dataset) assert ds.is_virtual assert ds.shape == (480, 255, 1024)
def test_run_get_virtual_dataset_filename(mock_fxe_raw_run, tmpdir): run = RunDirectory(mock_fxe_raw_run) path = str(tmpdir / 'test-vds.h5') ds = run.get_virtual_dataset('FXE_DET_LPD1M-1/DET/6CH0:xtdf', 'image.data', filename=path) assert_isfile(path) assert ds.file.filename == path assert isinstance(ds, h5py.Dataset) assert ds.is_virtual assert ds.shape == (61440, 1, 256, 256)