code: def _testBasic(self, dtype):
      x = np.arange(200, dtype=np.float32).astype(dtype)
      np.random.shuffle(x)
  
      # Check that argmin and argmax match numpy along the primary axis
      self._testBothArg(math_ops.argmax, x, 0, x.argmax())
      self._testBothArg(math_ops.argmin, x, 0, x.argmin())
  
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 3 for calling the ArgMaxTest obj's underlying member method code with example usage: obj._testBasic(dtype) without return types