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()