code: def get_chunks(tests, modulo, mode):
      chunks = []
      if mode == "MODULO":
          for modulo_index in range(modulo):
              chunks.append(
                  test_splitter.get_shuffled_chunk(
                      tests, modulo, modulo_index
                  )
              )
      elif mode == "SEQUENTIAL":
          for modulo_index in range(modulo):
              chunks.append(
                  test_splitter.get_sequential_chunk(
                      tests, modulo, modulo_index
                  )
              )
      else:
          raise ValueError("no such mode")
      return chunks
  
file_path: catboost/library/python/testing/yatest_lib/tests/test_testsplitter.py
imports: from yatest_lib import test_splitter
label_desc: Use image node_id 1 for calling a global function with example usage: get_chunks(tests, modulo, mode) and returns: chunks