code: def testKeyValueSort(self):
      for key_type in self._supported_key_types():
          for value_type in self._supported_key_types():
              if key_type == np.uint8 or value_type == np.uint8:
                  # I do not understand why the test fails on uint8. We plan to
                  # deprecate xla.key_value_sort in favor of xla.variadic_sort anyway.
                  continue
              x = self._shuffled_arange((101,), key_type)
              y = (-x).astype(value_type)
              self._assertOpOutputMatchesExpected(
                  xla.key_value_sort,
                  [x, y],
                  expected=[
                      np.arange(101, dtype=key_type),
                      -np.arange(101, dtype=value_type),
                  ],
              )
  
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 5 for calling the XlaSortOpTest obj's underlying member method code with example usage: obj.testKeyValueSort() without return types