code: def testArgMaxArgMin(self):
      data = [
          0,
          5,
          [1],
          [1, 2, 3],
          [[1, 2, 3]],
          [[4, 6], [7, 8]],
          [[[4, 6], [9, 10]], [[7, 8], [12, 34]]],
      ]
      for fn, d in itertools.product(self.array_transforms, data):
          arr = fn(d)
          # argmax and argmin returns indices (int64 type).
          self.match(
              np_math_ops.argmax(arr), np.argmax(arr), check_dtype=False
          )
          self.match(
              np_math_ops.argmin(arr), np.argmin(arr), check_dtype=False
          )
          if hasattr(arr, "shape"):
              ndims = len(arr.shape)
          else:
              ndims = np_array_ops.array(arr, copy=False).ndim
          if ndims == 0:
              # Numpy flattens the scalar ndarray and treats it as a 1-d array of
              # size 1.
              ndims = 1
          for axis in range(-ndims, ndims):
              self.match(
                  np_math_ops.argmax(arr, axis=axis),
                  np.argmax(arr, axis=axis),
                  check_dtype=False,
              )
              self.match(
                  np_math_ops.argmin(arr, axis=axis),
                  np.argmin(arr, axis=axis),
                  check_dtype=False,
              )
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import flexible_dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import test_util
  from tensorflow.python.framework import weak_tensor
  from tensorflow.python.ops import weak_tensor_ops
  from tensorflow.python.ops import weak_tensor_test_util
  from tensorflow.python.ops.numpy_ops import np_array_ops
  from tensorflow.python.ops.numpy_ops import np_arrays
  from tensorflow.python.ops.numpy_ops import np_math_ops
  from tensorflow.python.ops.numpy_ops import np_random
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 9 for calling the MathTest obj's underlying member method code with example usage: obj.testArgMaxArgMin() without return types