def test_no_reference_with_detrend(self): "Tests with no reference but detrend" output = NormalizeSignal(self.signal, fs=self.fs, detrend=True) test_output = FilterSignal(self.signal, highcut=40.0, fs=self.fs) test_output = DeltaFOverF(test_output) test_output = test_output - FilterSignal(test_output, btype='savgol') equal = np.array_equal(test_output, output) self.assertTrue(equal)
def test_with_reference_no_detrend(self): "Tests with reference and no detrend." output = NormalizeSignal(self.signal, self.reference, fs=self.fs, detrend=False) test_sig = FilterSignal(self.signal, highcut=40.0, fs=self.fs) test_ref = FilterSignal(self.reference, highcut=40.0, fs=self.fs) test_sig = DeltaFOverF(test_sig) test_ref = DeltaFOverF(test_ref) test_output = test_sig - test_ref equal = np.array_equal(test_output, output) self.assertTrue(equal)
def test_no_reference_nor_detrend(self): "Tests with no reference and no detrend." output = NormalizeSignal(self.signal, fs=self.fs, detrend=False) test_output = FilterSignal(self.signal, highcut=40.0, fs=self.fs) test_output = DeltaFOverF(test_output) equal = np.array_equal(test_output, output) self.assertTrue(equal)
def test_savgol_works(self): "Tests savgol functionality" signal = FilterSignal(self.signal, savgol_order=self.savgol_order, window_length=self.window_length, btype='savgol') test_signal = ssp.savgol_filter(self.signal, self.window_length, self.savgol_order, axis=0) equal = np.array_equal(signal, test_signal) self.assertTrue(equal)
def test_bandstop_works(self): "Tests bandpass functionality" signal = FilterSignal(self.signal, lowcut=self.lowcut, highcut=self.highcut, fs=self.fs, order=self.order, btype='bandstop') b, a = ButterFilterDesign(lowcut=self.lowcut, highcut=self.highcut, fs=self.fs, order=self.order, btype='bandstop') test_signal = ssp.filtfilt(b, a, self.signal) equal = np.array_equal(signal, test_signal) self.assertTrue(equal)