def get_peak_data():
""" Function for getting some pregenerated physio data """
physio = io.load_physio(get_test_data_path('ECG.csv'), fs=1000)
filt = operations.filter_physio(physio, [5., 15.], 'bandpass')
peaks = operations.peakfind_physio(filt)
return peaks
def test_plot_physio():
for data in [NOFS, WITHFS]:
assert isinstance(operations.plot_physio(data), matplotlib.axes.Axes)
peaks = operations.peakfind_physio(WITHFS)
assert isinstance(operations.plot_physio(peaks), matplotlib.axes.Axes)
fig, ax = plt.subplots(1, 1)
assert ax == operations.plot_physio(peaks, ax=ax)
def test_delete_peaks():
to_delete = [24685, 44169]
peaks = operations.peakfind_physio(WITHFS)
deleted = operations.delete_peaks(peaks, to_delete)
assert len(deleted.peaks) == len(peaks.peaks) - len(to_delete)
def test_reject_peaks():
to_reject = [24685, 44169]
peaks = operations.peakfind_physio(WITHFS)
rejected = operations.reject_peaks(peaks, to_reject)
assert len(rejected.peaks) == len(peaks.peaks) - len(to_reject)
def test_peakfind_physio():
with pytest.raises(ValueError):
operations.peakfind_physio(NOFS)
operations.peakfind_physio(NOFS, dist=20)
operations.peakfind_physio(NOFS, thresh=0.4, dist=20)
operations.peakfind_physio(WITHFS)
operations.peakfind_physio(WITHFS, dist=20)
operations.peakfind_physio(WITHFS, thresh=0.4)