code: def test_clustered_subset_array_shape():
      """[Base] ClusteredSubsetIndex: test the array shape on generation."""
      t = list()
      e = list()
      for tri, tei in ClusteredSubsetIndex(cl, 2, 2, X=X).generate(
          as_array=True
      ):
          t.append(tri.tolist())
          e.append(tei.tolist())
  
      assert t == [[0], [1], [2, 3], [4]]
      assert e == [[1, 2, 3, 4], [0, 2, 3, 4], [0, 1, 4], [0, 1, 2, 3]]
  
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 37 for calling a global function with example usage: test_clustered_subset_array_shape() without return types