def test_run_get_dataframe(mock_fxe_raw_run): run = RunDirectory(mock_fxe_raw_run) df = run.get_dataframe(fields=[("*_XGM/*", "*.i[xy]Pos*")]) assert len(df.columns) == 4 assert "SA1_XTD2_XGM/DOOCS/MAIN/beamPosition.ixPos" in df.columns df2 = run.get_dataframe(fields=[("*_XGM/*", "*.i[xy]Pos*")], timestamps=True) assert len(df2.columns) == 8 assert "SA1_XTD2_XGM/DOOCS/MAIN/beamPosition.ixPos" in df2.columns assert "SA1_XTD2_XGM/DOOCS/MAIN/beamPosition.ixPos.timestamp" in df2.columns
from time import monotonic from extra_data import RunDirectory print("Opening raw run...") start = monotonic() run = RunDirectory('/gpfs/exfel/exp/SA1/201830/p900025/raw/r0150/') delta = monotonic() - start print(len(run.files), "files") print(delta, "seconds") print() print("Retrieving data frame for XGM ixPos & iyPos...") start = monotonic() df = run.get_dataframe(fields=[("*_XGM/*", "*.i[xy]Pos"), ("*_XGM/*", "*.photonFlux")]) delta = monotonic() - start print(delta, "seconds") print(df.head())