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)
