code: def _get_dctn_test_axes(shape):
      axes = [[]]
      ndims = len(shape)
      axes.append(None)
      for naxes in range(1, min(ndims, 3) + 1):
          axes.extend(itertools.combinations(range(ndims), naxes))
      for index in range(1, ndims + 1):
          axes.append((-index,))
      return axes
  
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 1 for calling a global function with example usage: _get_dctn_test_axes(shape) and returns: axes