def test_run_single_train_from_index(): test_run = RunDirectory(RUNPATH) tid, data = test_run.train_from_index(541) assert len(data) == 2 tid, data = test_run.train_from_index(123) assert len(data) == 11 img = data['SPB_DET_AGIPD1M-1/DET/9CH0:xtdf']['image.data'] assert img.shape == (60, 512, 128) with pytest.raises(IndexError) as info: tid, data = test_run.train_from_index(542) print(info)
def test_train_from_index_fxe_run(mock_fxe_run): run = RunDirectory(mock_fxe_run) _, data = run.train_from_index(479) assert 'FXE_DET_LPD1M-1/DET/15CH0:xtdf' in data assert 'image.data' in data['FXE_DET_LPD1M-1/DET/15CH0:xtdf'] assert 'FXE_XAD_GEC/CAM/CAMERA' in data assert 'firmwareVersion.value' in data['FXE_XAD_GEC/CAM/CAMERA']