From 777f1c33a32d4dce6595dbc0279da218ffd3c636 Mon Sep 17 00:00:00 2001
From: kienle <kienle@passcal.nmt.edu>
Date: Tue, 27 Sep 2022 17:07:32 -0600
Subject: [PATCH] Add tests for checkWFChan
---
tests/test_model/test_handling_data.py | 44 +++++++++++++++++++++++++-
1 file changed, 43 insertions(+), 1 deletion(-)
diff --git a/tests/test_model/test_handling_data.py b/tests/test_model/test_handling_data.py
index e2ed926fd..b16ef6c2a 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
--
GitLab