Function test_csd

mne/time_frequency/tests/test_csd.py:53–105

Test constructing a CrossSpectralDensity.

()

def test_csd():
    """Test constructing a CrossSpectralDensity."""
    csd = CrossSpectralDensity(
        [1, 2, 3], ["CH1", "CH2"], frequencies=1, n_fft=1, tmin=0, tmax=1
    )
    assert_array_equal(csd._data, [[1], [2], [3]])  # Conversion to 2D array
    assert_array_equal(csd.frequencies, [1])  # Conversion to 1D array

    # Channels don't match
    raises(
        ValueError,
        CrossSpectralDensity,
        [1, 2, 3],
        ["CH1", "CH2", "Too many!"],
        tmin=0,
        tmax=1,
        frequencies=1,
        n_fft=1,
    )
    raises(
        ValueError,
        CrossSpectralDensity,
        [1, 2, 3],
        ["too little"],
        tmin=0,
        tmax=1,
        frequencies=1,
        n_fft=1,
    )

    # Frequencies don't match
    raises(
        ValueError,
        CrossSpectralDensity,
        [[1, 2], [3, 4], [5, 6]],
        ["CH1", "CH2"],
        tmin=0,
        tmax=1,
        frequencies=1,
        n_fft=1,
    )

    # Invalid dims
    raises(
        ValueError,
        CrossSpectralDensity,
        [[[1]]],
        ["CH1"],
        frequencies=1,
        n_fft=1,
        tmin=0,
        tmax=1,
    )


def test_csd_repr():