def test_run_info(capsys):
test_run = RunDirectory(RUNPATH)
test_run.info()
captured = capsys.readouterr()
print(captured)
assert captured[0] == ('Run information\n'
'\tDuration: 0:08:57.200000\n'
'\tFirst train ID: 1472806005\n'
'\tLast train ID: 1472811377\n'
'\t# of trains: 542\n'
'\n'
'Devices\n'
'\tInstruments\n'
'\t- SPB_DET_AGIPD1M-1/DET/0CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/10CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/1CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/2CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/3CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/4CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/5CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/6CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/7CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/8CH0:xtdf\n'
'\t- SPB_DET_AGIPD1M-1/DET/9CH0:xtdf\n'
'\tControls\n'
'\t-\n')
def load_data(run, proposal, exp, VERBOSE=False):
ddir = "/gpfs/exfel/exp/SPB/{}/{}/raw/{}/".format(exp, proposal, run)
print("Loading data from: {}".format(ddir))
data = RunDirectory(ddir)
if VERBOSE:
data.info()
return data
def test_read_spb_proc_run(mock_spb_proc_run):
run = RunDirectory(mock_spb_proc_run) #Test for calib data
assert len(run.files) == 16 # only 16 detector modules for calib data
assert run.train_ids == list(range(10000, 10064)) #64 trains
tid, data = next(run.trains())
device = 'SPB_DET_AGIPD1M-1/DET/15CH0:xtdf'
assert tid == 10000
for prop in ('image.gain', 'image.mask', 'image.data'):
assert prop in data[device]
assert 'u1' == data[device]['image.gain'].dtype
assert 'u4' == data[device]['image.mask'].dtype
assert 'f4' == data[device]['image.data'].dtype
run.info() # Smoke test
def test_read_fxe_run(mock_fxe_run):
run = RunDirectory(mock_fxe_run)
assert len(run.files) == 18 # 16 detector modules + 2 control data files
assert [tid for tid, _ in run.ordered_trains] == list(range(10000, 10480))
run.info() # Smoke test
def test_read_fxe_raw_run(mock_fxe_raw_run):
run = RunDirectory(mock_fxe_raw_run)
assert len(run.files) == 18 # 16 detector modules + 2 control data files
assert run.train_ids == list(range(10000, 10480))
run.info() # Smoke test
