Function test_plot_csd

mne/viz/tests/test_misc.py:332–366

Test plotting of CSD matrices.

(ch_types)

    ],
)
def test_plot_csd(ch_types):
    """Test plotting of CSD matrices."""
    if isinstance(ch_types, list):
        n_ch_types = len(ch_types)
        n_ch = 2 * n_ch_types
        ch_types = np.repeat(ch_types, 2).tolist()
    else:
        n_ch = 2
    ch_names = [f"CH{i + 1}" for i in range(n_ch)]
    n_data = n_ch * (n_ch + 1) // 2

    csd = CrossSpectralDensity(
        np.arange(1, n_data + 1),
        ch_names,
        frequencies=[(10, 20)],
        n_fft=1,
        tmin=0,
        tmax=1,
    )

    if ch_types is None:
        info = None
        expected_n_figs = 1
    else:
        info = create_info(ch_names, sfreq=1.0, ch_types=ch_types)
        unique_types = set(ch_types) if isinstance(ch_types, list) else {ch_types}
        expected_n_figs = len(unique_types)

    for mode in ("csd", "coh"):
        if ch_types == "misc":
            with pytest.raises(RuntimeError, match="No plottable channel types"):
                plot_csd(csd, info=info, mode=mode, show=False)
        else:
            figs = plot_csd(csd, info=info, mode=mode, show=False)
            assert len(figs) == expected_n_figs


@pytest.mark.slowtest  # Slow on Azure