code: def test_index_mapping_one_table(self):
      num_sc = 4
      x, shards = create_test_table_shards(
          TestTable(vocab=12, dim=4, shift=0), num_sc
      )
      indices = tf_constant([1, 2, 5, 7, 9])
      shard_idx, position_in_shard = v3_utils.map_indices_in_shard(
          num_sparse_cores=num_sc,
          offset_in_shard=0,
          shard_rotation=0,
          row_indices=indices,
      )
      self.assertAllEqual(
          shard_idx,
          indices % num_sc,
      )
      self.assertAllEqual(
          [x[i] for i in indices],
          [shards[j][k] for j, k in zip(shard_idx, position_in_shard)],
      )
  
file_path: tensorflow-upstream/tensorflow/python/tpu/tpu_embedding_v3_utils_test.py
imports: import collections
  from absl.testing import parameterized
  from tensorflow.python.compat import v2_compat
  from tensorflow.python.framework.constant_op import constant
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
  from tensorflow.python.tpu import tpu_embedding_v3_utils
label_desc: Use image node_id 5 for calling the TpuEmbeddingV3UtilsTest obj's underlying member method code with example usage: obj.test_index_mapping_one_table() without return types