From 4c322603d323d18a9d773592d1773f7db11bf547 Mon Sep 17 00:00:00 2001 From: linarphy Date: Mon, 27 May 2024 16:34:49 +0200 Subject: [PATCH] Fix edge case --- src/science_signal/signal.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/science_signal/signal.py b/src/science_signal/signal.py index e0d2953..d6500d0 100644 --- a/src/science_signal/signal.py +++ b/src/science_signal/signal.py @@ -74,7 +74,7 @@ class Signal: start = min(self.x) if end is None: end = max(self.x) - indexes = where(logical_and(self.x > start, self.x < end)) + indexes = where(logical_and(self.x >= start, self.x <= end)) return Signal( self.x[indexes], self.y[indexes], @@ -120,7 +120,7 @@ class Signal: end = max(abs(freq_x)) index_to_remove = where( - logical_and(abs(freq_x) < start, abs(freq_x) > end) + logical_and(abs(freq_x) <= start, abs(freq_x) >= end) ) freq_y[index_to_remove] = 0 y = irfft(freq_y)