code: 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 _record_warnings():  # n_fft sometimes
              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)
  
file_path: mne-python/mne/time_frequency/tests/test_stockwell.py
imports: from pathlib import Path
  import numpy
  import pytest
  from numpy.testing import assert_allclose, assert_array_almost_equal, assert_array_less, assert_equal
  from scipy import fftpack
  from mne import Epochs, make_fixed_length_events, read_events
  from mne.io import read_raw_fif
  from mne.time_frequency import AverageTFR, tfr_array_stockwell
  from mne.time_frequency._stockwell import _check_input_st, _precompute_st_windows, _st, _st_power_itc, tfr_stockwell
  from mne.utils import _record_warnings
label_desc: Use image node_id 2 for calling a global function with example usage: test_stockwell_check_input() without return types