add more personality to the test

SpikeInterface · Jun 20, 2024 · be59dbe · be59dbe
1 parent 3dbb8c1
commit be59dbe
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/spikeinterface/preprocessing/tests/test_scaling.py b/src/spikeinterface/preprocessing/tests/test_scaling.py
@@ -15,10 +15,11 @@ def test_scale_to_uv():
         sampling_frequency=sampling_frequency,
     )
 
-    gains = np.ones(shape=(num_channels))
-    offsets = np.zeros(shape=(num_channels))
-    recording.set_channel_gains(gains)  # Random gains
-    recording.set_channel_offsets(offsets)  # Random offsets
+    rng = np.random.default_rng(0)
+    gains = rng.random(size=(num_channels)).astype(np.float32)
+    offsets = rng.random(size=(num_channels)).astype(np.float32)
+    recording.set_channel_gains(gains)
+    recording.set_channel_offsets(offsets)
 
     # Apply the preprocessor
     scaled_recording = ScaleTouV(recording=recording)