def test_baseline_subtractor(camera_waveforms): waveforms, _ = camera_waveforms n_pixels, _ = waveforms.shape rand = np.random.RandomState(1) offset = np.arange(n_pixels)[:, np.newaxis] waveforms = rand.normal(0, 0.1, waveforms.shape) + offset assert_allclose(waveforms[3].mean(), 3, rtol=1e-2) baseline_subtracted = subtract_baseline(waveforms, 0, 10) assert_allclose(baseline_subtracted.mean(), 0, atol=1e-3)
def test_baseline_subtractor(camera_waveforms): waveforms, _ = camera_waveforms n_chan, n_pixels, n_samples = waveforms.shape rand = np.random.RandomState(1) offset = np.arange(n_pixels)[np.newaxis, :, np.newaxis] waveforms = rand.normal(0, 0.1, waveforms.shape) + offset assert_allclose(waveforms[0, 3].mean(), 3, rtol=1e-2) baseline_subtracted = subtract_baseline(waveforms, 0, 10) assert_allclose(baseline_subtracted.mean(), 0, atol=1e-3)