From 6dc1f72469624f0c2fe5326a407e8db97cd103dc Mon Sep 17 00:00:00 2001 From: JoeZiminski Date: Fri, 15 Nov 2024 11:55:20 +0000 Subject: [PATCH] Fix tests. --- src/spikeinterface/preprocessing/tests/test_whiten.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/spikeinterface/preprocessing/tests/test_whiten.py b/src/spikeinterface/preprocessing/tests/test_whiten.py index 22297839f1..e3d0d6c2e3 100644 --- a/src/spikeinterface/preprocessing/tests/test_whiten.py +++ b/src/spikeinterface/preprocessing/tests/test_whiten.py @@ -131,7 +131,7 @@ def test_compute_covariance_matrix(self, dtype): otherwise it can overflow. """ eps = 1e-16 - _, cov_mat, recording = self.get_test_recording(num_segments=1, dtype=dtype) + _, cov_mat, recording = self.get_test_recording(dtype=dtype) whitened_recording = whiten( recording, @@ -161,7 +161,7 @@ def test_non_default_eps(self): the cov mat if the correct eps is used. """ eps = 1 - _, cov_mat, recording = self.get_test_recording(num_segments=1, dtype=np.float32) + _, cov_mat, recording = self.get_test_recording(dtype=np.float32) whitened_recording = whiten( recording, @@ -218,7 +218,7 @@ def test_apply_mean(self, apply_mean): means = np.array([10, 20, 30]) eps = 1e-16 - _, cov_mat, recording = self.get_test_recording(num_segments=1, dtype=np.float32, means=means) + _, cov_mat, recording = self.get_test_recording(dtype=np.float32, means=means) whitened_recording = whiten( recording, @@ -269,7 +269,7 @@ def test_whiten_regularisation_norm(self): whitening preprocessing is the same as the one computed from sklearn when regularise kwargs are given. """ - _, _, recording = self.get_test_recording(num_segments=1, dtype=np.float32) + _, _, recording = self.get_test_recording(dtype=np.float32) whitened_recording = whiten( recording, @@ -297,7 +297,7 @@ def test_local_vs_global_whiten(self): channels are considered for whitening. Test that whitening is correct for the first pair and last pair. """ - _, _, recording = self.get_test_recording(num_segments=1, dtype=np.float32) + _, _, recording = self.get_test_recording(dtype=np.float32) y_dist = 2 recording.set_channel_locations(