diff --git a/tests/test_model/test_handling_data.py b/tests/test_model/test_handling_data.py index e2ed926fd26ecf5247b11bee24d2246c03a29521..b16ef6c2aa0f0d574a86545b1c4a2df19fbb047f 100644 --- a/tests/test_model/test_handling_data.py +++ b/tests/test_model/test_handling_data.py @@ -644,4 +644,46 @@ class TestCheckSohChan(TestCase): def test_channel_not_requested(self): for channel_id in self.sample_channel_ids: - self.assertFalse(checkSOHChan(channel_id, self.req_soh_chans)) \ No newline at end of file + self.assertFalse(checkSOHChan(channel_id, self.req_soh_chans)) + + +class TestCheckWfChan(TestCase): + def setUp(self) -> None: + self.req_wf_chans = ['LHE', 'HHE'] + self.sample_channel_ids = ['LHE', 'HHE', 'AL2', 'MNZ', 'VNN'] + + def test_all_channels_requested(self): + self.req_wf_chans = ['*'] + with self.subTest('test_waveform_channel'): + for channel_id in self.sample_channel_ids: + self.assertTupleEqual( + checkWFChan(channel_id, self.req_wf_chans), + ('WF', True) + ) + with self.subTest('test_non_waveform_channel'): + channel_id = 'Not a waveform channel' + self.assertTupleEqual( + checkWFChan(channel_id, self.req_wf_chans), + ('', True) + ) + + def test_channel_is_requested(self): + channel_id = 'LHE' + self.assertTupleEqual( + checkWFChan(channel_id, self.req_wf_chans), + ('WF', True) + ) + + def test_channel_not_requested(self): + with self.subTest('test_waveform_channel'): + channel_id = 'AHE' + self.assertTupleEqual( + checkWFChan(channel_id, self.req_wf_chans), + ('WF', False) + ) + with self.subTest('test_non_waveform_channel'): + channel_id = 'Not a waveform channel' + self.assertTupleEqual( + checkWFChan(channel_id, self.req_wf_chans), + ('', False) + ) \ No newline at end of file