code: def test_distributed_work_chunk_size(
      mpi_size, num_live_points_missing
  ):
      processes = range(mpi_size)
      todo = [
          distributed_work_chunk_size(
              num_live_points_missing, rank, mpi_size
          )
          for rank in processes
      ]
      assert sum(todo) == num_live_points_missing
      assert max(todo) - min(todo) in {0, 1}
  
file_path: ultranest/tests/test_utils.py
imports: import numpy
  import tempfile
  import os
  from ultranest.utils import vectorize, is_affine_transform, normalised_kendall_tau_distance, make_run_dir
  from ultranest.utils import distributed_work_chunk_size
  from numpy.testing import assert_allclose
  import pytest
label_desc: Use image node_id 5 for calling a global function with example usage: test_distributed_work_chunk_size(mpi_size, num_live_points_missing) without return types