mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2026-09-21 07:22:09 +02:00
Erik Fröjdhandmazzol_a
0dc72634a9
We have no code in aare to deal with RawFiles that have missing packets if they are not padded to full size or frames with packet loss dropped. Added: - check that we don't try to read files that are not supported - python bindings for frame discard policy at least we get the check in: closes #171 --------- Co-authored-by: AliceMazzoleni99 <alice.mazzoleni@psi.ch>
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
|
|
|
|
import json
|
|
|
|
import pytest
|
|
from aare import RawMasterFile, ReadoutMode, DetectorType, FrameDiscardPolicy
|
|
|
|
|
|
@pytest.mark.withdata
|
|
def test_read_rawfile_quad_eiger_and_compare_to_numpy(test_data_path):
|
|
|
|
file_name = test_data_path/'raw/jungfrau/jungfrau_single_master_0.json'
|
|
|
|
f = RawMasterFile(file_name)
|
|
assert(f.reading_mode == ReadoutMode.UNKNOWN)
|
|
assert(f.detector_type == DetectorType.Jungfrau)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"policy, expected_policy",
|
|
[
|
|
("nodiscard", FrameDiscardPolicy.NoDiscard),
|
|
("discard", FrameDiscardPolicy.Discard),
|
|
("discardpartial", FrameDiscardPolicy.DiscardPartial),
|
|
],
|
|
)
|
|
def test_raw_master_file_context_manager(tmp_path, policy, expected_policy):
|
|
file_name = tmp_path / "run_master_0.json"
|
|
file_name.write_text(json.dumps({
|
|
"Version": 7.2,
|
|
"Detector Type": "Jungfrau",
|
|
"Timing Mode": "auto",
|
|
"Geometry": {"x": 1, "y": 1},
|
|
"Image Size in bytes": 12,
|
|
"Pixels": {"x": 3, "y": 2},
|
|
"Max Frames Per File": 1,
|
|
"Total Frames": 2,
|
|
"Frames in File": 2,
|
|
"Frame Padding": 1,
|
|
"Frame Discard Policy": policy,
|
|
}))
|
|
|
|
|
|
with RawMasterFile(file_name) as context_file:
|
|
assert context_file.reading_mode == ReadoutMode.UNKNOWN
|
|
assert context_file.detector_type == DetectorType.Jungfrau
|
|
frame_discard_policy = context_file.frame_discard_policy
|
|
assert isinstance(frame_discard_policy, FrameDiscardPolicy)
|
|
assert frame_discard_policy == expected_policy
|