"Added tests for SingleNodeScheduler"

noisepy · Oct 5, 2023 · 0ec07a0 · 0ec07a0
1 parent 8b97fb3
commit 0ec07a0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/noisepy/seis/noise_module.py b/src/noisepy/seis/noise_module.py
@@ -14,7 +14,7 @@
 
 # import pycwt
 import scipy
-from numba import jit,float32
+from numba import jit
 from obspy.core.inventory import Channel, Inventory, Network, Site, Station
 from obspy.core.util.base import _get_function_from_entry_point
 from obspy.signal.filter import bandpass
@@ -1257,7 +1257,7 @@ def portion_gaps(stream, starttime: obspy.UTCDateTime, endtime: obspy.UTCDateTim
     return pgaps
 
 
-@jit("float32[:](float32[:],float32)",nopython=True)
+@jit("float32[:](float32[:],float32)")
 def segment_interpolate(sig1, nfric):
     """
     this function interpolates the data to ensure all points located on interger times of the

diff --git a/tests/test_scheduler.py b/tests/test_scheduler.py
@@ -1,5 +1,6 @@
 from noisepy.seis.scheduler import SingleNodeScheduler
 
+
 def test_single_node_scheduler_get_indices():
     single_node_scheduler = SingleNodeScheduler()
     items = ["a", "b", "c", "d"]
@@ -8,8 +9,9 @@ def test_single_node_scheduler_get_indices():
 
     assert indices == expected_indices
 
+
 def test_synchronize():
     scheduler = SingleNodeScheduler()
     result = scheduler.synchronize()
 
-    assert result is None
+    assert result is None