def test_detect_slowwave_Ngo2015(): detsw = DetectSlowWave(method='Ngo2015') detsw.invert = True assert repr(detsw) == 'detsw_Ngo2015_0.50-1.20Hz' sw = detsw(data) assert len(sw.events) == 4
def test_detect_slowwave_AASM_Massimini2004(): detsw = DetectSlowWave(method='AASM/Massimini2004') detsw.invert = True assert repr(detsw) == 'detsw_AASM/Massimini2004_0.10-4.00Hz' sw = detsw(data) assert len(sw.events) == 15
def test_detect_slowwave_to_data(): detsw = DetectSlowWave() detsw.invert = True sw = detsw(data) sw_data = sw.to_data('count') assert sw_data(0)[1] == 1 sw_ptp = sw.to_data('ptp') assert approx(sw_ptp(0)[1]) == 63.0
def test_detect_slowwave_to_data(): detsw = DetectSlowWave() sw = detsw(data) sw_data = sw.to_data('count') assert sw_data(0)[1] == 2 sw_auc = sw.to_data('area_under_curve') assert approx(sw_auc(0)[1]) == -16.454917436089122 sw_ptp = sw.to_data('ptp') assert approx(sw_ptp(0)[1]) == 84.0
""" out, _, _ = detect_Moelle2011(data, sf, time, opts_spin) indices = np.zeros((len(out), 2)) for i, k in enumerate(out): indices[i, 0] = k['start'] indices[i, 1] = k['end'] indices *= sf return indices.astype(int) ############################################################################### # Slow-waves function # ~~~~~~~~~~~~~~~~~~~ # Define a DetectSlowWave instance : opts_sw = DetectSlowWave('Massimini2004') # Define the function to replace : def fcn_slowwave(data, sf, time, hypno): # noqa """New slowwave detection function. See : https://wonambi-python.github.io/api/wonambi.detect.slowwave.html for an exhaustive list of implemented detections inside wonambi. """ out = detect_Massimini2004(data, sf, time, opts_sw) indices = np.zeros((len(out), 2)) for i, k in enumerate(out): indices[i, 0] = k['start'] indices[i, 1] = k['end'] indices *= sf
def test_detect_slowwave_Massimini2004(): detsw = DetectSlowWave() assert repr(detsw) == 'detsw_Massimini2004_0.10-4.00Hz' sw = detsw(data) assert len(sw.events) == 2