Python _check_input_stの例

コード例 #1

ファイル: test_stockwell.py プロジェクト: 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)

コード例 #2

ファイル: test_stockwell.py プロジェクト: 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)

コード例 #3

ファイル: test_stockwell.py プロジェクト: 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)