code: def test_equal_split(shutdown_only):
      ray.init(num_cpus=2)
  
      def range2x(n):
          return ray.data.range(2 * n)
  
      def counts(shards):
          @ray.remote(num_cpus=0)
          def count(s):
              return s.count()
  
          return ray.get([count.remote(s) for s in shards])
  
      r1 = counts(range2x(10).split(3, equal=True))
      assert all(c == 6 for c in r1), r1
  
imports: import itertools
  import math
  import random
  import time
  from unittest.mock import patch
  import numpy
  import pandas
  import pytest
  import ray
  from ray.data._internal.block_list import BlockList
  from ray.data._internal.equalize import _equalize
  from ray.data._internal.execution.interfaces import RefBundle
  from ray.data._internal.logical.interfaces import LogicalPlan
  from ray.data._internal.logical.operators.input_data_operator import InputData
  from ray.data._internal.plan import ExecutionPlan
  from ray.data._internal.split import _drop_empty_block_split, _generate_global_split_results, _generate_per_block_split_indices, _generate_valid_indices, _split_at_indices, _split_single_block
  from ray.data._internal.stats import DatasetStats
  from ray.data.block import BlockAccessor, BlockMetadata
  from ray.data.dataset import Dataset
  from ray.data.tests.conftest import *
  from ray.data.tests.util import extract_values
  from ray.tests.conftest import *
label_desc: Use image node_id 1 for calling a global function with example usage: test_equal_split(shutdown_only) and returns: ray, ray, s