code: def test_split_forward_backward(seed, axis, ctx, func_name):
      from nbla_test_utils import function_tester
  
      rng = np.random.RandomState(seed)
      shape = [2, 3, 4]
      x = rng.randn(*shape).astype(np.float32)
      inputs = [x]
      function_tester(
          rng,
          F.split,
          ref_split,
          inputs,
          func_args=[axis],
          ctx=ctx,
          func_name=func_name,
          atol_b=1e-2,
      )
  
file_path: nnabla/python/test/function/test_split.py
imports: import pytest
  import numpy
  import nnabla
  import nnabla.functions
  from nbla_test_utils import list_context
label_desc: Use image node_id 2 for calling a global function with example usage: test_split_forward_backward(seed, axis, ctx, func_name) without return types