code: def testOneHot(self):
      data = np.array([[1, 2], [1, 3], [2, 2], [1, 1]], dtype=np.int32)
      encoded = dataset_utilities._one_hot(data)
      expected = [
          [1, 0, 0, 1, 0],
          [1, 0, 0, 0, 1],
          [0, 1, 0, 1, 0],
          [1, 0, 1, 0, 0],
      ]
      np.testing.assert_array_equal(encoded, expected)
  
file_path: agents/tf_agents/bandits/environments/dataset_utilities_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  import tensorflow
  from tf_agents.bandits.environments import dataset_utilities
label_desc: Use image node_id 1 for calling the DatasetUtilitiesTest obj's underlying member method code with example usage: obj.testOneHot() without return types