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)
