From f71c978b13957cb99a25b3dcfeff1d7ad9aac701 Mon Sep 17 00:00:00 2001
From: Jeremy Fee <jmfee@usgs.gov>
Date: Mon, 16 Dec 2019 16:17:13 -0700
Subject: [PATCH] Split get_delta_from_interval to separate method

---
 geomagio/TimeseriesUtility.py | 39 ++++++++++++++++++++++++++---------
 1 file changed, 29 insertions(+), 10 deletions(-)

diff --git a/geomagio/TimeseriesUtility.py b/geomagio/TimeseriesUtility.py
index 046c10d3..05db4e2c 100644
--- a/geomagio/TimeseriesUtility.py
+++ b/geomagio/TimeseriesUtility.py
@@ -35,16 +35,7 @@ def create_empty_trace(starttime, endtime, observatory,
     obspy.core.Trace
         trace for the requested channel
     """
-    if interval == 'tenhertz':
-        delta = 0.1
-    elif interval == 'second':
-        delta = 1.
-    elif interval == 'minute':
-        delta = 60.
-    elif interval == 'hour':
-        delta = 3600.
-    elif interval == 'day':
-        delta = 86400.
+    delta = get_delta_from_interval(interval)
     stats = obspy.core.Stats()
     stats.network = network
     stats.station = station
@@ -62,6 +53,34 @@ def create_empty_trace(starttime, endtime, observatory,
     return obspy.core.Trace(data, stats)
 
 
+def get_delta_from_interval(data_interval):
+    """convert interval name to number of seconds
+
+    Parameters
+    ----------
+    interval : str
+        interval length {day, hour, minute, second, tenhertz}
+
+    Returns
+    -------
+    int
+        number of seconds for interval, or None if unknown
+    """
+    if data_interval == 'tenhertz':
+        delta = 0.1
+    elif data_interval == 'second':
+        delta = 1.
+    elif data_interval == 'minute':
+        delta = 60.
+    elif data_interval == 'hour':
+        delta = 3600.
+    elif data_interval == 'day':
+        delta = 86400.
+    else:
+        delta = None
+    return delta
+
+
 def get_stream_start_end_times(timeseries, without_gaps=False):
     """get start and end times from a stream.
             Traverses all traces, and find the earliest starttime, and
-- 
GitLab