code: def _get_dctn_test_s(shape, axes):
      s_list = [None]
      if axes is not None:
          s_list.extend(
              itertools.product(
                  *[
                      [
                          shape[ax] + i
                          for i in range(-shape[ax] + 1, shape[ax] + 1)
                      ]
                      for ax in axes
                  ]
              )
          )
      return s_list
  
file_path: jax/tests/scipy_fft_test.py
imports: import itertools
  from absl.testing import absltest
  from jax._src import test_util
  import jax.scipy.fft
  import scipy.fft
  from jax import config
label_desc: Use image node_id 2 for calling a global function with example usage: _get_dctn_test_s(shape, axes) and returns: s_list