

def test_waveform_amplitude(masks, waveforms):
    waveforms *= 0.1
    masks *= 0.1
    waveforms[:, 10, :] *= 10
    masks[:, 10] *= 10
    mean_waveforms = mean(waveforms)
    mean_masks = mean(masks)
    amplitude = get_waveform_amplitude(mean_masks, mean_waveforms)
    assert np.all((amplitude >= 0))
    assert (amplitude.shape == (mean_waveforms.shape[1],))

dataset/ETHPy150Open kwikteam/phy/phy/stats/tests/test_clusters.py test_waveform_amplitude/original