From aff4cb227cd2bb857c3e0903e3c3e2710bd05ab7 Mon Sep 17 00:00:00 2001 From: appel_c Date: Fri, 1 Mar 2024 13:15:14 +0100 Subject: [PATCH] refactor: fix _add_noise --- ophyd_devices/sim/sim_data.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ophyd_devices/sim/sim_data.py b/ophyd_devices/sim/sim_data.py index c2584e5..932d2ca 100644 --- a/ophyd_devices/sim/sim_data.py +++ b/ophyd_devices/sim/sim_data.py @@ -456,9 +456,9 @@ class SimulatedDataMonitor(SimulatedDataBase): v = np.random.poisson(v) return v elif noise == NoiseType.UNIFORM: - noise = np.random.uniform(-1, 1) * noise_multiplier - v += noise - return v if v > 0 else v + noise = np.ceil(np.random.uniform(0, 1) * noise_multiplier).astype(int) + v += noise * (np.random.randint(0, 2) * 2 - 1) + return v if v > 0 else 0 return v