def test_misaligned_correction(peaks_misaligned, artifacts_misaligned): peaks_corrected = _correct_artifacts(artifacts_misaligned, peaks_misaligned) assert np.unique( peaks_corrected ).size == peaks_misaligned.size # make sure that no peak duplication occurs and that number of peaks doesn't change
def test_extra_correction(peaks_extra, artifacts_extra): peaks_corrected = _correct_artifacts(artifacts_extra, peaks_extra) assert np.unique(peaks_corrected).size == (peaks_extra.size - len(artifacts_extra["extra"]))
def test_missed_correction(peaks_missed, artifacts_missed): peaks_corrected = _correct_artifacts(artifacts_missed, peaks_missed) assert np.unique(peaks_corrected).size == (peaks_missed.size + len(artifacts_missed["missed"]))