Python RunDirectory.train_from_index Examples

Example #1

File: test_reader.py Project: Landau1908/karabo_data

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)

Example #2

File: test_reader_mockdata.py Project: Landau1908/karabo_data

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']