def test_thres_extrema(): rs, = argrelextrema(data1[:, 0], lambda x, y: thres_extrema(x, y, 0.1)) assert len(rs) == 1 assert rs == 1