code: def get_splitted_tests(
      test_entities,
      modulo,
      modulo_index,
      partition_mode,
      is_sorted=False,
  ):
      if partition_mode == "SEQUENTIAL":
          return get_sequential_chunk(
              test_entities, modulo, modulo_index, is_sorted
          )
      elif partition_mode == "MODULO":
          return get_shuffled_chunk(
              test_entities, modulo, modulo_index, is_sorted
          )
      else:
          raise ValueError(
              "detected unknown partition mode: {}".format(
                  partition_mode
              )
          )
  
imports: import collections
label_desc: Use image node_id 4 for calling a global function with example usage: get_splitted_tests(test_entities, modulo, modulo_index, partition_mode, is_sorted) and returns: get_sequential_chunk, get_shuffled_chunk