code: def test_split(x, indices_or_sections, teardown):
      x1 = Input()
      ys = Split(indices_or_sections, axis=0)(x1)
      model = Model(x1, ys)
  
      y_expected = np.split(x, indices_or_sections, axis=0)
      y_pred = model.predict(x)
      y_pred = listify(y_pred)
  
      for actual, expected in safezip2(y_pred, y_expected):
          assert_array_equal(actual, expected)
  
file_path: baikal/tests/steps/test_merge.py
imports: import numpy
  import pytest
  from numpy.testing import assert_array_equal
  from baikal import Model, Input
  from baikal._core.utils import listify, safezip2
  from baikal.steps import ColumnStack, Concatenate, Stack, Split
  from tests.helpers.fixtures import teardown
label_desc: Use image node_id 4 for calling a global function with example usage: test_split(x, indices_or_sections, teardown) without return types