code: def testInputShardingWithGeneratePerSplit(self):
      # 25, 5, 11 train, dev, test input shards
      # 9 output tasks in all (2 + 3 + 4), so
      #
      # Division should be like:
      #
      # Train
      # task_id 0 -> 0, .. 12
      # task_id 1 -> 13 .. 24
      #
      # Dev
      # task_id 2 -> 0, 1
      # task_id 3 -> 2, 3,
      # task_id 4 -> 4
      #
      # Test
      # task_id 5 -> 0, 1, 2
      # task_id 6 -> 3, 4, 5
      # task_id 7 -> 6, 7, 8
      # task_id 8 -> 9, 10
  
      expected_input_file_sharding = [
          ["train-%05d-of-00025" % i for i in range(13)],  # task_id 0
          [
              "train-%05d-of-00025" % i for i in range(13, 25)
          ],  # task_id 1
          ["dev-%05d-of-00005" % i for i in [0, 1]],  # task_id 2
          ["dev-%05d-of-00005" % i for i in [2, 3]],  # task_id 3
          ["dev-%05d-of-00005" % i for i in [4]],  # task_id 4
          ["test-%05d-of-00011" % i for i in [0, 1, 2]],  # task_id 5
          ["test-%05d-of-00011" % i for i in [3, 4, 5]],  # task_id 6
          ["test-%05d-of-00011" % i for i in [6, 7, 8]],  # task_id 7
          ["test-%05d-of-00011" % i for i in [9, 10]],  # task_id 8
      ]
  
      problem = FakeDistributedProblem()
  
      list_input_files = []
      for task_id in range(9):
          input_files = problem._task_id_to_input_files(task_id)
          list_input_files.append(
              [
                  os.path.basename(input_file)
                  for input_file in input_files
              ]
          )
  
      self.assertSequenceEqual(
          expected_input_file_sharding, list_input_files
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import shutil
  from tensor2tensor.data_generators import problem
  from tensor2tensor.data_generators import text_encoder
  from tensor2tensor.data_generators import text_problems
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
label_desc: Use image node_id 4 for calling the DistributedText2TextProblemsTest obj's underlying member method code with example usage: obj.testInputShardingWithGeneratePerSplit() without return types