improve testings

EtienneCmb · EtienneCmb · commit 6644cde88549 · 2020-08-03T14:22:03.000+02:00
diff --git a/frites/simulations/sim_distant_mi.py b/frites/simulations/sim_distant_mi.py
@@ -141,4 +141,4 @@ def sim_gauss_fit(stim_type='cont_linear', n_epochs=400, n_sti=4, n_pts=400,
     # add random noise to remove equal values for gcmi
     x = x + rnd_x.rand(*x.shape) / 100.
     y = y + rnd_y.rand(*x.shape) / 100.
-    return x.T, y.T, stim
+    return x.T, y.T, stim.squeeze()
diff --git a/frites/simulations/tests/test_sim_distant_mi.py b/frites/simulations/tests/test_sim_distant_mi.py
@@ -0,0 +1,29 @@
+"""Test simulating distant MI."""
+import numpy as np
+
+from frites.simulations import (sim_gauss_fit, sim_distant_cc_ms,
+                                sim_distant_cc_ss)
+
+
+class TestDistantMI(object):
+
+    def test_sim_gauss_fit(self):
+        for stim_type in ['discrete_stim', 'cont_linear', 'cont_flat']:
+            x, y, stim = sim_gauss_fit(stim_type=stim_type, n_epochs=20)
+            assert x.shape == y.shape
+            if stim_type is not 'discrete_stim':
+                assert len(stim) == x.shape[0]
+
+    def test_sim_distant_cc_ss(self):
+        n_epochs = 20
+        x, y, roi = sim_distant_cc_ss(n_epochs=n_epochs)
+        assert x.shape == (len(y), len(roi), 400)
+
+    def test_sim_distant_cc_ms(self):
+        n_subjects = 3
+        x, y, roi, times = sim_distant_cc_ms(n_subjects)
+        assert len(x) == len(y) == len(roi) == n_subjects
+        assert [k.shape == x[0].shape for k in x]
+        assert [k.shape == y[0].shape for k in y]
+        assert [k.shape == roi[0].shape for k in roi]
+        assert x[0].shape == (len(y[0]), len(roi[0]), len(times))
diff --git a/frites/stats/stats_param.py b/frites/stats/stats_param.py
@@ -54,7 +54,7 @@ def rfx_ttest(mi, mi_p, center=False, zscore=False, ttested=False):
     mi_p : array_like
         A list of array of permuted mutual information of shape
         (n_perm, n_suj, n_times). If `ttested` is True, n_suj shoud be 1.
-    center : {'mean', "median", "trimmed"} | False
+    center : {'mean', 'median', 'trimmed'} | False
         If True, substract the mean of the surrogates to the true and permuted
         mi. The median or the 20% trimmed mean can also be removed
         :cite:`wilcox2018guide`
diff --git a/frites/stats/tests/test_stats_param.py b/frites/stats/tests/test_stats_param.py
@@ -24,9 +24,10 @@ def test_rfx_ttest(self):
         assert tv.shape == (n_roi, n_times)
         assert tv_p.shape == (n_perm, n_roi, n_times)
         # center
-        tv, tv_p = rfx_ttest(x, x_p, center=True)
-        assert tv.shape == (n_roi, n_times)
-        assert tv_p.shape == (n_perm, n_roi, n_times)
+        for center in [False, 'mean', 'median', 'trimmed']:
+            tv, tv_p = rfx_ttest(x, x_p, center=center)
+            assert tv.shape == (n_roi, n_times)
+            assert tv_p.shape == (n_perm, n_roi, n_times)
         # zscore
         tv, tv_p = rfx_ttest(x, x_p, zscore=True)
         assert tv.shape == (n_roi, n_times)
diff --git a/frites/utils/tests/test_perf.py b/frites/utils/tests/test_perf.py
@@ -0,0 +1,27 @@
+"""Test performance tools."""
+import numpy as np
+from time import sleep
+
+from frites.utils.perf import timeit, id, get_data_base, arrays_share_data
+
+
+class TestPerfTools(object):
+
+    def test_timeit(self):
+        @timeit
+        def fcn(sec): return sleep(sec)  # noqa
+
+    def test_id(self):
+        x = np.random.rand(1000)
+        assert id(x) != id(x.copy())
+        assert id(x) == id(x)
+
+    def test_get_data_base(self):
+        x = np.random.rand(1000)
+        np.testing.assert_array_almost_equal(x, get_data_base(x))
+
+    def test_arrays_share_data(self):
+        x = np.random.rand(1000)
+        y = np.random.rand(1000)
+        assert not arrays_share_data(x, y)
+        assert arrays_share_data(x, x)
diff --git a/frites/workflow/tests/test_wf_mi.py b/frites/workflow/tests/test_wf_mi.py
@@ -30,7 +30,11 @@ class TestWfMi(object):  # noqa
 
     def test_definition(self):
         """Test workflow definition."""
-        WfMi(mi_type='cc', inference='rfx')
+        y, gt = sim_mi_cc(x, snr=1.)
+        dt = DatasetEphy(x, y, roi, times=time)
+        wf = WfMi(mi_type='cc', inference='rfx')
+        wf.fit(dt, **kw_mi)
+        wf.tvalues
 
     def test_mi_cc(self):
         """Test method fit."""
@@ -93,3 +97,13 @@ def test_no_stat(self):
         wf.fit(dt, mcp='maxstat', **kw_mi)
         t_end_2 = tst()
         assert t_end_1 - t_start_1 > t_end_2 - t_start_2
+
+    def test_conjunction_analysis(self):
+        """Test the conjunction analysis."""
+        y, gt = sim_mi_cc(x, snr=1.)
+        dt = DatasetEphy(x, y, roi, times=time)
+        wf = WfMi(mi_type='cc', inference='rfx')
+        mi, pv = wf.fit(dt, **kw_mi)
+        cj_ss, cj = wf.conjunction_analysis(dt)
+        assert cj_ss.shape == (n_subjects, n_times, n_roi)
+        assert cj.shape == (n_times, n_roi)
