code: def test_one_hot(self):
      indices = random_ops.random_uniform(
          [3, 2, 3], minval=0, maxval=4, dtype=dtypes.int32
      )
  
      def loop_fn(i):
          indices_i = array_ops.gather(indices, i)
          return (
              array_ops.one_hot(
                  indices_i, depth=4, on_value=2.0, off_value=-2.0
              ),
              array_ops.one_hot(indices_i, depth=4, axis=1),
          )
  
      self._test_loop_fn(loop_fn, 3)
  
imports: from tensorflow.python.eager import backprop
  from tensorflow.python.framework import constant_op
  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 array_ops_stack
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import tensor_array_grad
  from tensorflow.python.ops.parallel_for import control_flow_ops
  from tensorflow.python.ops.parallel_for.test_util import PForTestCase
  from tensorflow.python.platform import test
label_desc: Use image node_id 12 for calling the ArrayTest obj's underlying member method code with example usage: obj.test_one_hot() and returns: array_ops,  array_ops