code: def test_full_array_shape():
      """[Base] FoldIndex: test the array shape on generation."""
      tr = [
          np.array([2, 3, 4]),
          np.array([0, 1, 4]),
          np.array([0, 1, 2, 3]),
      ]
      te = [np.array([0, 1]), np.array([2, 3]), np.array([4])]
  
      for i, (tri, tei) in enumerate(
          FoldIndex(3, X).generate(as_array=True)
      ):
          np.testing.assert_array_equal(tri, tr[i])
          np.testing.assert_array_equal(tei, te[i])
  
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 8 for calling a global function with example usage: test_full_array_shape() without return types