From 6f9391c6ecde51511921899423faab12879b296d Mon Sep 17 00:00:00 2001
From: kienle <kienle@passcal.nmt.edu>
Date: Thu, 29 Sep 2022 15:45:32 -0600
Subject: [PATCH] Add tests for trim_downsample_SOHChan
---
tests/test_model/test_handling_data.py | 119 ++++++++++++++++++++++++-
1 file changed, 118 insertions(+), 1 deletion(-)
diff --git a/tests/test_model/test_handling_data.py b/tests/test_model/test_handling_data.py
index e387c4fc5..75864f6b9 100644
--- a/tests/test_model/test_handling_data.py
+++ b/tests/test_model/test_handling_data.py
@@ -33,7 +33,8 @@ from sohstationviewer.model.handling_data import (
sortData,
squash_gaps,
downsample,
- chunk_minmax
+ chunk_minmax,
+ trim_downsample_SOHChan
)
from sohstationviewer.model.reftek.from_rt2ms.core import Reftek130
@@ -854,3 +855,119 @@ class TestChunkMinmax(TestCase):
times, data = chunk_minmax(self.times, self.data, req_points)
self.assertEqual(times.size, 0)
self.assertEqual(data.size, 0)
+
+
+class TestTrimDownsampleSohChan(TestCase):
+ @staticmethod
+ def downsample(times, data, req_points):
+ return times, data
+
+ def setUp(self) -> None:
+ self.channel_info = {}
+ self.org_trace = {
+ 'times': np.arange(1000),
+ 'data': np.arange(1000)
+ }
+ self.channel_info['orgTrace'] = self.org_trace
+ self.start_time = 250
+ self.end_time = 750
+ self.first_time = False
+
+ patcher = patch('sohstationviewer.model.handling_data.downsample')
+ self.addCleanup(patcher.stop)
+ self.mock_downsample = patcher.start()
+ self.mock_downsample.side_effect = self.downsample
+
+ def num_points_outside_time_range(self, start_time, end_time):
+ return len([data_point
+ for data_point in self.org_trace['times']
+ if not start_time <= data_point <= end_time])
+
+ def test_mseed_start_time_later_than_times_data(self):
+ self.start_time = 250
+ self.end_time = 1250
+ trim_downsample_SOHChan(self.channel_info, self.start_time,
+ self.end_time, self.first_time)
+ self.assertGreaterEqual(self.channel_info['times'].min(),
+ self.start_time)
+ self.assertEqual(
+ self.org_trace['times'].size - self.channel_info['times'].size,
+ self.num_points_outside_time_range(self.start_time, self.end_time)
+ )
+
+ def test_mseed_end_time_earlier_than_times_data(self):
+ self.start_time = -250
+ self.end_time = 750
+ trim_downsample_SOHChan(self.channel_info, self.start_time,
+ self.end_time, self.first_time)
+ self.assertLessEqual(self.channel_info['times'].max(),
+ self.end_time)
+ self.assertEqual(
+ self.org_trace['times'].size - self.channel_info['times'].size,
+ self.num_points_outside_time_range(self.start_time, self.end_time)
+ )
+
+ def test_mseed_start_time_earlier_than_times_data(self):
+ self.start_time = -250
+ self.end_time = 750
+ trim_downsample_SOHChan(self.channel_info, self.start_time,
+ self.end_time, self.first_time)
+ self.assertEqual(self.channel_info['times'].min(),
+ self.channel_info['orgTrace']['times'].min())
+ self.assertEqual(
+ self.org_trace['times'].size - self.channel_info['times'].size,
+ self.num_points_outside_time_range(self.start_time, self.end_time)
+ )
+
+ def test_mseed_end_time_later_than_times_data(self):
+ self.start_time = 250
+ self.end_time = 1250
+ trim_downsample_SOHChan(self.channel_info, self.start_time,
+ self.end_time, self.first_time)
+ self.assertEqual(self.channel_info['times'].max(),
+ self.channel_info['orgTrace']['times'].max())
+ self.assertEqual(
+ self.org_trace['times'].size - self.channel_info['times'].size,
+ self.num_points_outside_time_range(self.start_time, self.end_time)
+ )
+
+ def test_mseed_times_data_contained_in_time_range(self):
+ self.start_time = -250
+ self.end_time = 1250
+ trim_downsample_SOHChan(self.channel_info, self.start_time,
+ self.end_time, self.first_time)
+ np.testing.assert_array_equal(self.channel_info['times'],
+ self.org_trace['times'])
+
+ def test_mseed_time_range_is_the_same_as_times_data(self):
+ self.start_time = 0
+ self.end_time = 999
+ trim_downsample_SOHChan(self.channel_info, self.start_time,
+ self.end_time, self.first_time)
+ np.testing.assert_array_equal(self.channel_info['times'],
+ self.org_trace['times'])
+
+ def test_mseed_time_range_does_not_overlap_times_data(self):
+ self.start_time = 2000
+ self.end_time = 3000
+ trim_downsample_SOHChan(self.channel_info, self.start_time,
+ self.end_time, self.first_time)
+ self.assertEqual(self.channel_info['times'].size, 0)
+ self.assertEqual(self.channel_info['data'].size, 0)
+
+ def test_mseed_data_is_downsampled(self):
+ trim_downsample_SOHChan(self.channel_info, self.start_time,
+ self.end_time, self.first_time)
+ self.assertTrue(self.mock_downsample.called)
+
+ def test_mseed_processed_data_is_stored_in_appropriate_location(self):
+ trim_downsample_SOHChan(self.channel_info, self.start_time,
+ self.end_time, self.first_time)
+ expected_keys = ('orgTrace', 'times', 'data')
+ self.assertTupleEqual(tuple(self.channel_info.keys()),
+ expected_keys)
+
+ @expectedFailure
+ def test_reftek(self):
+ self.fail()
+
--
GitLab