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