From a18654094eaf7f0458de721983af4e0d0b36aa51 Mon Sep 17 00:00:00 2001
From: Maeva Pourpoint <maeva@passcal.nmt.edu>
Date: Thu, 9 Sep 2021 09:37:03 -0600
Subject: [PATCH] Update update_stats_... method names
---
lemi2seed/lemi_data.py | 8 ++++----
tests/test_lemi_data.py | 6 +++---
2 files changed, 7 insertions(+), 7 deletions(-)
diff --git a/lemi2seed/lemi_data.py b/lemi2seed/lemi_data.py
index b129756..ff6d674 100644
--- a/lemi2seed/lemi_data.py
+++ b/lemi2seed/lemi_data.py
@@ -213,7 +213,7 @@ class LemiData():
samples)], dtype=dtype)
ind += 1
- def update_stats_1(self) -> None:
+ def update_stats_geo_time(self) -> None:
"""
Update stats attribute by defining the latitude, longitude, elevation,
start and end time for all channels.
@@ -226,7 +226,7 @@ class LemiData():
self.stats['time_period_start'] = self.data["Time_stamp"][0]
self.stats['time_period_end'] = self.data["Time_stamp"][-1]
- def update_stats_2(self, net: str, sta: str) -> None:
+ def update_stats_net_sta(self, net: str, sta: str) -> None:
"""
Update stats attribute by defining or redefining the network code and
station name.
@@ -273,7 +273,7 @@ class LemiData():
data_channels = str2list(metadata.station.components_recorded)
e_info = {x.channel_number: x.component for x in metadata.electric
if x.channel_number in data_channels}
- self.update_stats_2(net, sta)
+ self.update_stats_net_sta(net, sta)
self.update_array(data_channels, e_info)
def prep_data(self):
@@ -284,4 +284,4 @@ class LemiData():
sys.exit(1)
else:
self.create_data_array()
- self.update_stats_1()
+ self.update_stats_geo_time()
diff --git a/tests/test_lemi_data.py b/tests/test_lemi_data.py
index 7efe51a..5b46cc7 100644
--- a/tests/test_lemi_data.py
+++ b/tests/test_lemi_data.py
@@ -174,11 +174,11 @@ class TestLemiData(unittest.TestCase):
self.assertEqual(lemi_data.stats['nbr_runs'], 4)
self.assertSequenceEqual(lemi_data.data_np[0], expected_data_np)
- def test_update_stats_1(self):
- """Test basic functionality of update_stats_1."""
+ def test_update_stats_geo_time(self):
+ """Test basic functionality of update_stats_geo_time."""
lemi_data = LemiData(self.path2data, OUTPUT_MSEED)
lemi_data.parse_all_files()
- lemi_data.update_stats_1()
+ lemi_data.update_stats_geo_time()
self.assertAlmostEqual(lemi_data.stats['latitude'], 34.04839, 5)
self.assertAlmostEqual(lemi_data.stats['longitude'], -107.12844, 5)
self.assertAlmostEqual(lemi_data.stats['elevation'], 2201.77122, 5)
--
GitLab