Fix platform-specific test fail

mne-tools · Jun 14, 2024 · 5e38316 · 5e38316
1 parent 0d9f501
commit 5e38316
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/mne_connectivity/decoding/tests/test_decomposition.py b/mne_connectivity/decoding/tests/test_decomposition.py
@@ -164,9 +164,9 @@ def test_spectral_decomposition(method, mode):
     epochs_transformed_2 = decomp_class_2.transform(
         X=epochs[: n_epochs // 2].get_data()
     )
-    assert_allclose(epochs_transformed, epochs_transformed_2, rtol=1e-5)
-    assert_allclose(decomp_class.filters_, decomp_class_2.filters_, rtol=1e-5)
-    assert_allclose(decomp_class.patterns_, decomp_class_2.patterns_, rtol=1e-5)
+    assert_allclose(epochs_transformed, epochs_transformed_2, atol=1e-9)
+    assert_allclose(decomp_class.filters_, decomp_class_2.filters_, atol=1e-9)
+    assert_allclose(decomp_class.patterns_, decomp_class_2.patterns_, atol=1e-9)
 
     # TEST FITTING ON ONE PIECE OF DATA AND TRANSFORMING ANOTHER
     con_mv_class_unseen_data = spectral_connectivity_epochs(
@@ -218,6 +218,9 @@ def test_spectral_decomposition(method, mode):
     decomp_class.set_params(rank=None)
 
 
+test_spectral_decomposition("cacoh", "cwt_morlet")
+
+
 @pytest.mark.parametrize("method", ["cacoh", "mic"])
 @pytest.mark.parametrize("mode", ["multitaper", "fourier", "cwt_morlet"])
 def test_spectral_decomposition_parallel(method, mode):