code: def test_batch_norm_invalid_dimensions(
      device,
      x_shape,
      gamma_shape,
      beta_shape,
      running_mean_shape,
      running_var_shape,
      axis,
      float_dtype,
  ):
      (
          x,
          gamma,
          beta,
          running_mean,
          running_var,
      ) = _create_batch_norm_ndarray_args(
          chainerx,
          device,
          x_shape,
          gamma_shape,
          beta_shape,
          running_mean_shape,
          running_var_shape,
          float_dtype,
      )
  
      with pytest.raises(chainerx.DimensionError):
          chainerx.batch_norm(
              x,
              gamma,
              beta,
              running_mean=running_mean,
              running_var=running_var,
              eps=1e-2,
              decay=0.9,
              axis=axis,
          )
  
imports: import unittest
  import chainer
  import numpy
  import pytest
  import chainerx
  import chainerx.testing
  from chainerx_tests import array_utils
  from chainerx_tests import op_utils
label_desc: Use image node_id 2 for calling a global function with example usage: test_batch_norm_invalid_dimensions(device, x_shape, gamma_shape, beta_shape, running_mean_shape, running_var_shape, axis, float_dtype) without return types