fix FIR zero filter for short length coefficients (#3881)

gwastro · Nov 29, 2021 · c0e2064 · c0e2064
1 parent a13c18e
commit c0e2064
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pycbc/filter/resample.py b/pycbc/filter/resample.py
@@ -52,7 +52,9 @@ def lfilter(coefficients, timeseries):
     # If there aren't many points just use the default scipy method
     if len(timeseries) < 2**7:
         series = scipy.signal.lfilter(coefficients, 1.0, timeseries)
-        return series
+        return TimeSeries(series,
+                          epoch=timeseries.start_time,
+                          delta_t=timeseries.delta_t)
     elif (len(timeseries) < fillen * 10) or (len(timeseries) < 2**18):
         cseries = (Array(coefficients[::-1] * 1)).astype(timeseries.dtype)
         cseries.resize(len(timeseries))

diff --git a/test/test_resample.py b/test/test_resample.py
@@ -82,7 +82,7 @@ def test_lfilter(self):
             test = test[len(c):]
 
             maxreldiff =  ((ref - test) / ref).max()
-
+            self.assertTrue(isinstance(test, TimeSeries))
             self.assertTrue(maxreldiff < 1e-7)
 
 suite = unittest.TestSuite()