def test_waveform_amplitude(masks, waveforms):
    waveforms *= .1
    masks *= .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 is (mean_waveforms.shape[1],)
dataset/ETHPy150Open kwikteam/phy/phy/stats/tests/test_clusters.py/test_waveform_amplitude/operator `==` -> `is`