Esempi in Python per _check_input_st

Esempio n. 1

File: test_stockwell.py Progetto: LiuFang816/SALSTM_py_data

def test_stockwell_check_input():
    """Test input checker for stockwell"""
    # check for data size equal and unequal to a power of 2

    for last_dim in (127, 128):
        data = np.zeros((2, 10, last_dim))
        x_in, n_fft, zero_pad = _check_input_st(data, None)

        assert_equal(x_in.shape, (2, 10, 128))
        assert_equal(n_fft, 128)
        assert_equal(zero_pad, 128 - last_dim)

Esempio n. 2

File: test_stockwell.py Progetto: deep-introspection/mne-python

def test_stockwell_check_input():
    """Test input checker for stockwell"""
    # check for data size equal and unequal to a power of 2

    for last_dim in (127, 128):
        data = np.zeros((2, 10, last_dim))
        x_in, n_fft, zero_pad = _check_input_st(data, None)

        assert_equal(x_in.shape, (2, 10, 128))
        assert_equal(n_fft, 128)
        assert_equal(zero_pad, 128 - last_dim)

Esempio n. 3

File: test_stockwell.py Progetto: tjochmann/mne-python

def test_stockwell_check_input():
    """Test input checker for stockwell."""
    # check for data size equal and unequal to a power of 2

    for last_dim in (127, 128):
        data = np.zeros((2, 10, last_dim))
        with warnings.catch_warnings(record=True):  # 127 < n_fft
            x_in, n_fft, zero_pad = _check_input_st(data, None)

        assert_equal(x_in.shape, (2, 10, 128))
        assert_equal(n_fft, 128)
        assert_equal(zero_pad, 128 - last_dim)