code: def _testDim(self, dtype):
      shape = (3, 2, 4, 5, 6, 3, 7)
      x = np.arange(
          functools.reduce(lambda x, y: x * y, shape), dtype=np.float32
      ).astype(dtype)
      np.random.shuffle(x)
      x = x.reshape(shape)
  
      # Check that argmin and argmax match numpy along all axes
      for axis in range(-7, 7):
          self._testBothArg(math_ops.argmax, x, axis, x.argmax(axis))
          self._testBothArg(math_ops.argmin, x, axis, x.argmin(axis))
  
imports: import functools
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 5 for calling the ArgMaxTest obj's underlying member method code with example usage: obj._testDim(dtype) without return types