From bb055bf2549d43ccfa7a0f9d467a264deaba67ac Mon Sep 17 00:00:00 2001 From: ldam <ldam@passcal.nmt.edu> Date: Tue, 22 Aug 2023 16:04:38 -0600 Subject: [PATCH] unittest for reading 2 mseed folders --- tests/model/mseed_data/test_mseed.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/tests/model/mseed_data/test_mseed.py b/tests/model/mseed_data/test_mseed.py index 2afa9b80c..e12494dcc 100644 --- a/tests/model/mseed_data/test_mseed.py +++ b/tests/model/mseed_data/test_mseed.py @@ -229,6 +229,32 @@ class TestMSeed(TestCase): self.assertEqual(len(obj.waveform_data['3203']['EL2']['tracesInfo']), 1) + def test_select_2_folders(self): + # is_multiplex = True => the selected channel will be read + args = { + 'data_type': 'Q330', + 'is_multiplex': True, + 'list_of_dir': [multiplex_data, blockettes_data], + 'req_soh_chans': [], + 'req_wf_chans': ['EL2'], + 'gap_minimum': 60 + } + obj = MSeed(**args) + self.assertEqual(list(obj.waveform_data.keys()), ['3203']) + self.assertEqual(list(obj.waveform_data['3203'].keys()), ['EL2']) + self.assertEqual(obj.waveform_data['3203']['EL2']['samplerate'], 200) + self.assertEqual(obj.waveform_data['3203']['EL2']['startTmEpoch'], + 1671730004.3100293) + self.assertEqual(obj.waveform_data['3203']['EL2']['endTmEpoch'], + 1671735657.9148998) + self.assertEqual(obj.waveform_data['3203']['EL2']['size'], 268576) + self.assertEqual(obj.waveform_data['3203']['EL2']['gaps'], + [[1671730720.5549, 1671735031.2799978]]) + self.assertEqual(len(obj.waveform_data['3203']['EL2']['tracesInfo']), + 1) + self.assertEqual(obj.gaps['3203'], + [[1671730720.5549, 1671735031.2799978]]) + def test_existing_time_range(self): # check if data_time is from the given range, end time may get # a little greater than read_end according to record's end time -- GitLab