diff --git a/tests/model/reftek_data/test_reftek.py b/tests/model/reftek_data/test_reftek.py index 8bb2a71749d59d30833d4c98014308fff58060df..af03b2aa03606d807d0e888b71a0091a0dcc9901 100644 --- a/tests/model/reftek_data/test_reftek.py +++ b/tests/model/reftek_data/test_reftek.py @@ -17,7 +17,7 @@ class TestReftek(TestCase): # raise exception when path not exist args = { 'data_type': 'RT130', - 'folder': '_', + 'list_of_dir': ['_'], 'rt130_waveform_data_req': False, 'on_unittest': True } @@ -31,7 +31,7 @@ class TestReftek(TestCase): def test_read_soh(self): args = { 'data_type': 'RT130', - 'folder': reftek_data, + 'list_of_dir': [reftek_data], 'req_soh_chans': [], 'rt130_waveform_data_req': False, 'on_unittest': True @@ -56,13 +56,13 @@ class TestReftek(TestCase): '\n150:00:00:00 REF TEK 130' '\r\n150:00:00:00 CPU SOFTWARE') self.assertEqual(list(obj.soh_data.keys()), [('92EB', '25')]) - self.assertEqual(list(obj.soh_data[('92EB', '25')].keys()), - expected_soh) + self.assertEqual(sorted(list(obj.soh_data[('92EB', '25')].keys())), + sorted(expected_soh)) def test_read_waveform(self): args = { 'data_type': 'RT130', - 'folder': reftek_data, + 'list_of_dir': [reftek_data], 'req_soh_chans': [], 'req_wf_chans': [1], 'rt130_waveform_data_req': True, @@ -85,7 +85,7 @@ class TestReftek(TestCase): def test_read_mass_pos(self): args = { 'data_type': 'RT130', - 'folder': reftek_data, + 'list_of_dir': [reftek_data], 'req_soh_chans': ['_'], 'include_mp123zne': True, 'rt130_waveform_data_req': False, @@ -107,7 +107,7 @@ class TestReftek(TestCase): with self.subTest("no gap_minimum set"): args = { 'data_type': 'RT130', - 'folder': reftek_gap_data, + 'list_of_dir': [reftek_gap_data], 'req_soh_chans': [], 'req_wf_chans': ['*'], 'rt130_waveform_data_req': True, @@ -129,7 +129,7 @@ class TestReftek(TestCase): with self.subTest("has gap_minimum set"): args = { 'data_type': 'RT130', - 'folder': reftek_gap_data, + 'list_of_dir': [reftek_gap_data], 'req_soh_chans': [], 'req_wf_chans': ['*'], 'rt130_waveform_data_req': True, @@ -148,3 +148,45 @@ class TestReftek(TestCase): ['TEXT', ('98AD', '0')]) self.assertEqual(obj.gaps[('98AD', '0')], [[1648493999.64, 1648508400.64]]) + + def test_select_2_folders(self): + args = { + 'data_type': 'RT130', + 'list_of_dir': [reftek_data, reftek_gap_data], + 'req_soh_chans': [], + 'req_wf_chans': ['*'], + 'rt130_waveform_data_req': True, + 'gap_minimum': 60, + 'on_unittest': True + } + expected_soh = [ + 'SOH/Data Def', 'Battery Volt', 'DAS Temp', 'Backup Volt', + 'Disk Usage1', 'Disk Usage2', 'Dump Called/Comp', 'GPS On/Off/Err', + 'GPS Lk/Unlk', 'Clk Phase Err', 'Event DS1', 'Event DS9'] + expected_waveform = ['DS1-1', 'DS1-2', 'DS1-3'] + obj = RT130(**args) + self.assertEqual(obj.found_data_streams, [9, 1, 1, 2, 2]) + self.assertEqual(obj.keys, [('92EB', '25'), ('98AD', '0')]) + self.assertEqual(obj.selected_key, ('92EB', '25')) + self.assertEqual( + list(obj.stream_header_by_key_chan[('92EB', '25')].keys()), + expected_waveform) + self.assertEqual(list(obj.log_data.keys()), + ['TEXT', ('92EB', '25'), ('98AD', '0')]) + self.assertEqual(len(obj.log_data['TEXT']), 0) + self.assertEqual(list(obj.log_data[('92EB', '25')].keys()), ['SOH']) + self.assertEqual(len(obj.log_data[('92EB', '25')]['SOH']), 1) + self.assertEqual( + obj.log_data[('92EB', '25')]['SOH'][0][:100], + '\nState of Health 17:150:00:00:00:000 ST: 92EB' + '\n150:00:00:00 REF TEK 130' + '\r\n150:00:00:00 CPU SOFTWARE') + self.assertEqual(list(obj.soh_data.keys()), + [('92EB', '25'), ('98AD', '0')]) + self.assertEqual(sorted(list(obj.soh_data[('92EB', '25')].keys())), + sorted(expected_soh)) + self.assertEqual(list(obj.gaps.keys()), + [('92EB', '25'), ('98AD', '0')]) + self.assertEqual(obj.gaps[('92EB', '25')], []) + self.assertEqual(obj.gaps[('98AD', '0')], + [[1648493999.64, 1648508400.64]])