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