code: def test_clustered_subset_partition_array():
      """[Base] ClusteredSubsetIndex: test partition indexing on arrays."""
      parts = list()
      for part in ClusteredSubsetIndex(cl, X=X).partition(
          as_array=True
      ):
          parts.append(part)
  
      np.testing.assert_array_equal(parts[0], np.array([0, 1]))
      np.testing.assert_array_equal(parts[1], np.array([2, 3, 4]))
  
imports: import os
  import subprocess
  import numpy
  from mlens import config
  from mlens.utils import IdTrain
  from mlens.index import FoldIndex, BlendIndex, SubsetIndex, TemporalIndex, ClusteredSubsetIndex, FullIndex
  from mlens.index.base import partition, prune_train
label_desc: Use image node_id 34 for calling a global function with example usage: test_clustered_subset_partition_array() without return types