From f45aa76bb13831709b293658c01fa35398209536 Mon Sep 17 00:00:00 2001
From: "Schwarz, Mario" <schmar9@gmx.de>
Date: Wed, 8 Jan 2025 10:39:55 +0100
Subject: [PATCH] added tests for llama decoder

---
 tests/conftest.py                        |   2 +-
 tests/llama/conftest.py                  |   8 ++
 tests/llama/test_llama_event_decoder.py  | 102 +++++++++++++++++++++++
 tests/llama/test_llama_header_decoder.py |  16 ++++
 tests/llama/test_llama_streamer.py       |  34 ++++++++
 5 files changed, 161 insertions(+), 1 deletion(-)
 create mode 100644 tests/llama/conftest.py
 create mode 100644 tests/llama/test_llama_event_decoder.py
 create mode 100644 tests/llama/test_llama_header_decoder.py
 create mode 100644 tests/llama/test_llama_streamer.py

diff --git a/tests/conftest.py b/tests/conftest.py
index aa8d49b..9104ae0 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -30,7 +30,7 @@ def pytest_sessionfinish(session, exitstatus):
 @pytest.fixture(scope="session")
 def lgnd_test_data():
     ldata = LegendTestData()
-    ldata.checkout("ecb370e")
+    #ldata.checkout("ecb370e")
     return ldata
 
 
diff --git a/tests/llama/conftest.py b/tests/llama/conftest.py
new file mode 100644
index 0000000..d3fbc2b
--- /dev/null
+++ b/tests/llama/conftest.py
@@ -0,0 +1,8 @@
+import fcutils
+import pytest
+
+
+# lgnd_test_data (LegendTestData) from fixture in root conftest.py
+@pytest.fixture(scope="module")
+def test_data_path(lgnd_test_data):
+    return lgnd_test_data.get_path("llamaDAQ/20241218-150158-pulser.bin")
\ No newline at end of file
diff --git a/tests/llama/test_llama_event_decoder.py b/tests/llama/test_llama_event_decoder.py
new file mode 100644
index 0000000..024a324
--- /dev/null
+++ b/tests/llama/test_llama_event_decoder.py
@@ -0,0 +1,102 @@
+from daq2lh5.llama.llama_streamer import LLAMAStreamer
+from daq2lh5.llama.llama_event_decoder import LLAMAEventDecoder, check_dict_spec_equal
+import pytest
+import lgdo
+
+def test_check_dict_spec_equal():
+    d1 = {"X": "1", "Y": "2", "Z": "3"}
+    d2 = {"X": "2", "Y": "2", "Z": "3"}
+    assert check_dict_spec_equal(d1, d2, ["Y", "Z"])
+    assert not check_dict_spec_equal(d1, d2, ["X", "Y"])
+
+@pytest.fixture(scope="module")
+def open_stream(test_data_path):
+    streamer = LLAMAStreamer()
+    streamer.open_stream(test_data_path)
+    yield streamer
+    streamer.close_stream()
+
+def test_get_key_lists(open_stream):
+    evt_dec : LLAMAEventDecoder = open_stream.event_decoder
+    assert evt_dec.get_key_lists() == [[0],[4]]
+
+def test_get_decoded_values(open_stream):
+    evt_dec : LLAMAEventDecoder = open_stream.event_decoder
+    dec_vals_0 = evt_dec.get_decoded_values(0)
+    assert dec_vals_0["waveform"]["wf_len"] == 2000
+    assert dec_vals_0["avgwaveform"]["wf_len"] == 10000
+    dec_vals_4 = evt_dec.get_decoded_values(4)
+    assert dec_vals_4["waveform"]["wf_len"] == 2000
+    assert dec_vals_4["avgwaveform"]["wf_len"] == 500
+
+
+def test_first_packet(open_stream):
+    good_packet = open_stream.read_packet()
+    assert good_packet
+    evt_dec: LLAMAEventDecoder = open_stream.event_decoder
+    assert evt_dec is not None
+    evt_rbkd = open_stream.event_rbkd
+    tbl = evt_rbkd[0].lgdo
+    assert isinstance(tbl, lgdo.Table)
+    ii = evt_rbkd[0].loc
+    assert ii == 1
+    ii = ii - 1 # use the last written entry (which is the only one, actually)
+    assert tbl["fch_id"].nda[ii] == 0
+    assert tbl["packet_id"].nda[ii] == 1
+    assert tbl["timestamp"].nda[ii] == 757530
+    assert tbl["peakHighValue"].nda[ii] == 9454
+    assert tbl["peakHighIndex"].nda[ii] == 1968
+    assert tbl["information"].nda[ii] == 0
+    assert tbl["accSum1"].nda[ii] == 7826
+    assert tbl["accSum2"].nda[ii] == 7826
+    assert tbl["accSum3"].nda[ii] == 7826
+    assert tbl["accSum4"].nda[ii] == 7826
+    assert tbl["accSum5"].nda[ii] == 7826
+    assert tbl["accSum6"].nda[ii] == 7826
+    assert tbl["accSum7"].nda[ii] == 7826
+    assert tbl["accSum8"].nda[ii] == 7826
+    assert tbl["waveform"]["dt"].nda[ii] > 3.999 and tbl["waveform"]["dt"].nda[ii] < 4.001
+    assert tbl["avgwaveform"]["dt"].nda[ii] > 15.999 and tbl["avgwaveform"]["dt"].nda[ii] < 16.001
+    assert tbl["waveform"]["t0"].nda[ii] > -4000.1 and tbl["waveform"]["t0"].nda[ii] < -3999.9
+    assert tbl["avgwaveform"]["t0"].nda[ii] > -8000.1 and tbl["avgwaveform"]["t0"].nda[ii] < -7999.9
+
+def test_first_packet_ch4(open_stream):
+    evt_rbkd = open_stream.event_rbkd
+    while True:
+        good_packet = open_stream.read_packet()
+        if not good_packet:
+            break
+    tbl = evt_rbkd[4].lgdo
+    assert evt_rbkd[4].loc > 0, "Not a single event of channel 4"
+    ii = 0
+    assert tbl["fch_id"].nda[ii] == 4
+    assert tbl["packet_id"].nda[ii] == 10
+    assert tbl["timestamp"].nda[ii] == 757530
+    assert tbl["peakHighValue"].nda[ii] == 7923
+    assert tbl["peakHighIndex"].nda[ii] == 371
+    assert tbl["information"].nda[ii] == 0
+    assert tbl["accSum1"].nda[ii] == 7912
+    assert tbl["accSum2"].nda[ii] == 7912
+    assert tbl["accSum3"].nda[ii] == 7912
+    assert tbl["accSum4"].nda[ii] == 7912
+    assert tbl["accSum5"].nda[ii] == 7912
+    assert tbl["accSum6"].nda[ii] == 7912
+    assert tbl["accSum7"].nda[ii] == 7912
+    assert tbl["accSum8"].nda[ii] == 7912
+    assert tbl["waveform"]["dt"].nda[ii] > 3.999 and tbl["waveform"]["dt"].nda[ii] < 4.001
+    assert tbl["avgwaveform"]["dt"].nda[ii] > 31.999 and tbl["avgwaveform"]["dt"].nda[ii] < 32.001
+    assert tbl["waveform"]["t0"].nda[ii] > -4000.1 and tbl["waveform"]["t0"].nda[ii] < -3999.9
+    assert tbl["avgwaveform"]["t0"].nda[ii] > -4000.1 and tbl["avgwaveform"]["t0"].nda[ii] < -3999.9
+
+def test_event_count(open_stream):
+    evt_rbkd = open_stream.event_rbkd
+    while True:
+        good_packet = open_stream.read_packet()
+        if not good_packet:
+            break
+    assert evt_rbkd[0].loc == 37
+    assert evt_rbkd[4].loc == 37
+
+
+
+
diff --git a/tests/llama/test_llama_header_decoder.py b/tests/llama/test_llama_header_decoder.py
new file mode 100644
index 0000000..fce819f
--- /dev/null
+++ b/tests/llama/test_llama_header_decoder.py
@@ -0,0 +1,16 @@
+
+from daq2lh5.llama.llama_header_decoder import LLAMAHeaderDecoder
+from daq2lh5.llama.llama_streamer import LLAMAStreamer
+
+def test_read_header(test_data_path):
+    streamer = LLAMAStreamer()
+    streamer.open_stream(test_data_path)
+    header = streamer.header_decoder
+    assert isinstance(header, LLAMAHeaderDecoder)
+    # following data is specific to the particular test file:
+    assert header.version_major == 2
+    assert header.version_minor == 0
+    assert header.version_patch == 0
+    assert header.length_econf == 88  # 22 words of 4 bytes
+    assert header.number_chOpen == 2
+    streamer.close_stream()
diff --git a/tests/llama/test_llama_streamer.py b/tests/llama/test_llama_streamer.py
new file mode 100644
index 0000000..16549a0
--- /dev/null
+++ b/tests/llama/test_llama_streamer.py
@@ -0,0 +1,34 @@
+from daq2lh5.llama.llama_streamer import LLAMAStreamer
+from daq2lh5.llama.llama_header_decoder import LLAMAHeaderDecoder
+from daq2lh5.llama.llama_event_decoder import LLAMAEventDecoder
+from daq2lh5.raw_buffer import RawBuffer
+
+def test_get_decoder_list():
+    streamer = LLAMAStreamer()
+    assert len(streamer.get_decoder_list()) == 2
+    assert isinstance(streamer.get_decoder_list()[0], LLAMAHeaderDecoder)
+    assert isinstance(streamer.get_decoder_list()[1], LLAMAEventDecoder)
+
+# test_data_path (str) from fixture in ./conftest.py
+def test_open_stream(test_data_path):
+    streamer = LLAMAStreamer()
+    rbl: list[RawBuffer] = streamer.open_stream(test_data_path)
+    assert len(rbl) == 1
+    assert isinstance(rbl[0], RawBuffer)
+    assert streamer.rb_lib is not None
+    nbytes_hdr = streamer.n_bytes_read
+    assert nbytes_hdr > 0
+    assert streamer.read_packet() # there has to be at last a single good packet
+    assert streamer.packet_id == 1
+    assert streamer.n_bytes_read > nbytes_hdr
+    streamer.close_stream()
+
+
+def test_open_stream_multiple(test_data_path):
+    streamer = LLAMAStreamer()
+    rbl: list[RawBuffer] = streamer.open_stream(test_data_path)
+    assert len(rbl) == 1
+    streamer.close_stream()
+    rbl: list[RawBuffer] = streamer.open_stream(test_data_path)
+    assert len(rbl) == 1
+    streamer.close_stream()
\ No newline at end of file