mirror of
https://github.com/paulscherrerinstitute/sf_daq_broker.git
synced 2026-05-03 18:14:13 +02:00
Dmitry Ozerov
43 lines
2.0 KiB
Python
43 lines
2.0 KiB
Python
import json
|
|
import unittest
|
|
|
|
from sf_databuffer_writer.utils import get_separate_writer_requests
|
|
|
|
|
|
class TestUtils(unittest.TestCase):
|
|
|
|
def test_get_separate_writer_requests(self):
|
|
start_pulse_id = 0
|
|
stop_pulse_id = 10
|
|
parameters = {"output_file": "test.h5"}
|
|
|
|
channels = ["channel_1", "channel_2"]
|
|
write_requests = list(get_separate_writer_requests(channels, parameters, start_pulse_id, stop_pulse_id))
|
|
self.assertEqual(len(write_requests), 1, "All non-camera channels should be bunched together.")
|
|
|
|
channels = ["channel_1", "camera:FPICTURE", "channel_2"]
|
|
write_requests = list(get_separate_writer_requests(channels, parameters, start_pulse_id, stop_pulse_id))
|
|
self.assertEqual(len(write_requests), 2, "The camera channel should be separate from the non-camera ones.")
|
|
|
|
channels = ["channel_1", "camera_1:FPICTURE", "channel_2", "camera_2:FPICTURE"]
|
|
write_requests = list(get_separate_writer_requests(channels, parameters, start_pulse_id, stop_pulse_id))
|
|
self.assertEqual(len(write_requests), 3, "Camera channels are one per request, bsread is bunched in one.")
|
|
|
|
bsread_channels_found = False
|
|
for write_request in write_requests:
|
|
channels = json.loads(write_request["data_api_request"])["channels"]
|
|
|
|
if len(channels) > 1:
|
|
self.assertFalse(bsread_channels_found, "There should be only 1 request with more than 1 channel.")
|
|
bsread_channels_found = True
|
|
|
|
for channel in channels:
|
|
self.assertFalse(channel["name"].endswith(":FPICTURE"), "There should be no image channels here.")
|
|
|
|
else:
|
|
self.assertTrue(channels[0]["name"].endswith(":FPICTURE"), "This channel should be a camera channel.")
|
|
self.assertEqual(json.loads(write_request["parameters"])["output_file"],
|
|
parameters["output_file"] + "_" + channels[0]["name"][:-9] + ".h5")
|
|
|
|
self.assertTrue(bsread_channels_found)
|