code: def testArgsort(self, dimension, dtype):
      shape = (2, 3, 4)
      if dtype not in self._supported_key_types():
          return
  
      def argsort(v, axis=dimension):
          return sort_ops.argsort(v, axis, stable=True)
  
      x = self._shuffled_arange(shape, dtype)
      self._assertOpOutputMatchesExpected(
          argsort,
          [x],
          expected=[np.argsort(x, axis=dimension, kind="stable")],
      )
  
imports: import unittest
  from absl.testing import parameterized
  import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.compiler.tf2xla.python import xla
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import function
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import sort_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 11 for calling the XlaSortOpTest obj's underlying member method code with example usage: obj.testArgsort(dimension, dtype) and returns: sort_ops