code: def test_splitrange():
      def f(n, k):
          assert _splitrange(n, k) == [
              len(a) for a in np.array_split(range(n), k)
          ]
  
      f(0, 1)
      f(5, 1)
      f(5, 3)
      f(5, 5)
      f(5, 10)
      f(50, 1)
      f(50, 2)
      f(50, 3)
      f(50, 4)
      f(50, 5)
  
file_path: ray/python/ray/data/tests/test_splitblocks.py
imports: import numpy
  import pytest
  import ray
  from ray.data._internal.execution.operators.map_transformer import _splitrange
  from ray.data.tests.conftest import *
  from ray.data.tests.conftest import CoreExecutionMetrics, assert_core_execution_metrics_equals, get_initial_core_execution_metrics_snapshot
  from ray.tests.conftest import *
label_desc: Use image node_id 1 for calling a global function with example usage: test_splitrange() without return types