code: def _test_sort(self, values, axis, direction):
      expected = np.sort(values, axis=axis)
      if direction == "DESCENDING":
          expected = np.flip(expected, axis=axis)
      self.assertAllEqual(
          expected,
          sort_ops.sort(
              constant_op.constant(values),
              axis=axis,
              direction=direction,
          ),
      )
  
imports: import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_util
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import sort_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the SortTest obj's underlying member method code with example usage: obj._test_sort(values, axis, direction) without return types