code: def test_distr_slice_axis(
      distr, slice_axis_args, expected_axis_length
  ):
      axis, begin, end = slice_axis_args
      distr_sliced = distr.slice_axis(axis, begin, end)
  
      assert distr_sliced.batch_shape[axis] == expected_axis_length
  
file_path: gluonts/test/mx/distribution/test_distribution_slice.py
imports: import mxnet
  import numpy
  import pytest
  import gluonts.mx.distribution.bijection
  from gluonts.mx.distribution import Beta, Binned, Dirichlet, Gamma, Laplace, MixtureDistribution, NegativeBinomial, PiecewiseLinear, Poisson, StudentT, TransformedDistribution, Uniform
  from gluonts.mx.distribution.box_cox_transform import BoxCoxTransform
  from gluonts.mx.distribution.gaussian import Gaussian
label_desc: Use image node_id 1 for calling a global function with example usage: test_distr_slice_axis(distr, slice_axis_args, expected_axis_length) without return types