code: def testOneHotOperation(self):
      x = mtf.zeros(self.mesh, self.ab_shape, dtype=tf.int32)
      one_hot_operation = mtf.OneHotOperation(
          x, self.c_dim, 1, 0, dtype=tf.bool
      )
      self.assertEqual(
          one_hot_operation.splittable_dims, frozenset(["a", "b", "c"])
      )
      self.assertEqual(one_hot_operation.unsplittable_dims, frozenset())
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import mesh_tensorflow
  from mesh_tensorflow import test_utils
  import numpy
  import tensorflow.compat.v1
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 18 for calling the OperationSplittabilityTest obj's underlying member method code with example usage: obj.testOneHotOperation() without return types