code: def test_chunk_method_for_valid_inputs(
      n: int, expected: List[List[int]]
  ) -> None:
      from hydra_plugins.hydra_ax_sweeper._core import CoreAxSweeper
  
      chunk_func = CoreAxSweeper.chunks
      batch = [1, 2, 3, 4, 5]
      out = list(chunk_func(batch, n))
      assert out == expected
  
imports: import math
  import os
  from pathlib import Path
  from typing import Any, List
  from hydra.core.plugins import Plugins
  from hydra.plugins.sweeper import Sweeper
  from hydra.test_utils.test_utils import TSweepRunner, chdir_plugin_root, run_python_script
  from omegaconf import DictConfig, OmegaConf
  from pytest import mark, raises
  from hydra_plugins.hydra_ax_sweeper.ax_sweeper import AxSweeper
label_desc: Use image node_id 3 for calling a global function with example usage: test_chunk_method_for_valid_inputs(n, expected) without return types