Add tests for checkWFChan

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