code: def get_shard_tests(args):
      test_names = list_tests(args.command[0])
      test_names = sorted(test_names)
  
      if args.fork_mode == "MODULO":
          return get_shuffled_chunk(
              test_names, args.split_factor, args.shard
          )
      elif args.fork_mode == "SEQUENTIAL":
          return get_sequential_chunk(
              test_names, args.split_factor, args.shard
          )
      else:
          raise ValueError(
              "detected unknown partition mode: {}".format(
                  args.fork_mode
              )
          )
  
file_path: catboost/build/scripts/split_unittest.py
imports: import argparse
  import tempfile
  import shlex
  import subprocess
label_desc: Use image node_id 5 for calling a global function with example usage: get_shard_tests(args) and returns: get_shuffled_chunk, get_sequential_chunk