code: def _testBothArg(
      self, method, x, axis, expected_values, expected_err_re=None
  ):
      self._testArg(
          method, x, axis, expected_values, True, expected_err_re
      )
      # Compilation time is too large with XLA/CPU autojit.
      if not test_util.is_xla_enabled():
          self._testArg(
              method, x, axis, expected_values, False, expected_err_re
          )
  
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 2 for calling the ArgMaxTest obj's underlying member method code with example usage: obj._testBothArg(method, x, axis, expected_values, expected_err_re) without return types