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)
