Improved time_frequency/tests/test_csd.py.

mne/time_frequency/tests/test_csd.py

@@ -16,21 +16,26 @@
                                 CrossSpectralDensity, read_csd,
                                 pick_channels_csd, psd_multitaper)
 from mne.time_frequency.csd import _sym_mat_to_vector, _vector_to_sym_mat
-from mne.proj import compute_proj_raw, Projection
+from mne.proj import Projection
 
 base_dir = op.join(op.dirname(__file__), '..', '..', 'io', 'tests', 'data')
 raw_fname = op.join(base_dir, 'test_raw.fif')
 event_fname = op.join(base_dir, 'test-eve.fif')
 
 
-def _make_csd():
+def _make_csd(add_proj=False):
     """Make a simple CrossSpectralDensity object."""
     frequencies = [1., 2., 3., 4.]
     n_freqs = len(frequencies)
     names = ['CH1', 'CH2', 'CH3']
     tmin, tmax = (0., 1.)
     data = np.arange(6. * n_freqs).reshape(n_freqs, 6).T
-    return CrossSpectralDensity(data, names, frequencies, 1, tmin, tmax)
+    if add_proj:
+        proj_data = dict(col_names=names, row_names=None, data=np.ones((1, len(names))))
+        projs = [Projection(data=proj_data)]
+    else:
+        projs = None
+    return CrossSpectralDensity(data, names, frequencies, 1, tmin, tmax, projs=projs)
 
 
 def test_csd():
@@ -224,15 +229,7 @@ def test_csd_get_data():
 @requires_version('h5io')
 def test_csd_save(tmp_path):
     """Test saving and loading a CrossSpectralDensity."""
-    raw = mne.io.read_raw_fif(raw_fname)
-    events = mne.find_events(raw)
-    picks = mne.pick_types(raw.info, meg='grad')
-    projs = compute_proj_raw(raw, n_grad=1, n_mag=1, n_eeg=1, reject=None)
-    raw.add_proj(projs)
-    epochs = mne.Epochs(raw, events, event_id=1, tmin=-0.2, tmax=1,
-                        picks=picks, baseline=(None, 0),
-                        reject=dict(grad=4000e-13), preload=True)
-    csd = csd_fourier(epochs, fmin=15, fmax=20)
+    csd = _make_csd(add_proj=True)
     fname = op.join(str(tmp_path), 'csd.h5')
     csd.save(fname)
     csd2 = read_csd(fname)