mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2026-09-21 10:52:10 +02:00
Erik Fröjdhandmazzol_a
b558f1146b
- seek to the edge of a file + read_into would select the wrong file - Visible for example in f.read_frame(index) in python Added fix, release notes and a small test. closes #371 Co-authored-by: AliceMazzoleni99 <alice.mazzoleni@psi.ch>
59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
|
|
|
|
import pytest
|
|
import numpy as np
|
|
from aare import CtbRawFile
|
|
|
|
|
|
@pytest.mark.withdata
|
|
def test_basic_properites(test_data_path):
|
|
f = CtbRawFile(test_data_path/'raw/ctb/run_master_0.json')
|
|
assert f.total_frames == 47
|
|
assert f.image_size_in_bytes == 640
|
|
assert f.master.analog_samples == 10
|
|
assert f.master.digital_samples is None
|
|
assert f.master.transceiver_samples is None
|
|
|
|
|
|
@pytest.mark.withdata
|
|
def test_seek_and_read(test_data_path):
|
|
f = CtbRawFile(test_data_path/'raw/ctb/run_master_0.json')
|
|
f.seek(0)
|
|
header, raw_data = f.read_frame()
|
|
assert header['frameNumber'] == 48 #we know that the first frame in this file is 48
|
|
|
|
#Seek a multiple of frames in file
|
|
f.seek(9)
|
|
header, raw_data = f.read_frame()
|
|
assert header['frameNumber'] == 57
|
|
|
|
@pytest.mark.withdata
|
|
def test_read_all_frames_direct_and_iterate(test_data_path):
|
|
"""
|
|
We want to make sure that iterating and reading give the same result.
|
|
"""
|
|
f0 = CtbRawFile(test_data_path/'raw/ctb/run_master_0.json')
|
|
f1 = CtbRawFile(test_data_path/'raw/ctb/run_master_0.json')
|
|
|
|
for h0, d0 in f0:
|
|
h1, d1 = f1.read_frame()
|
|
assert h0 == h1
|
|
assert np.all(d0 == d1)
|
|
|
|
@pytest.mark.withdata
|
|
def test_read_all_frames_with_index_and_iterate(test_data_path):
|
|
"""
|
|
We want to make sure that iterating and reading give the same result.
|
|
"""
|
|
f0 = CtbRawFile(test_data_path/'raw/ctb/run_master_0.json')
|
|
f1 = CtbRawFile(test_data_path/'raw/ctb/run_master_0.json')
|
|
|
|
frames_read = 0
|
|
for i,(h0, d0) in enumerate(f0):
|
|
h1, d1 = f1.read_frame(i)
|
|
assert h0 == h1
|
|
assert np.all(d0 == d1)
|
|
frames_read += 1
|
|
|
|
assert frames_read == f0.total_frames
|