code: def test_array_input(self):
      # Input: an 4+1+3=8-length array
      # Expected output: a list of 3 arrays (4-, 1-, and 3-length)
      in_size = 2
      seqs_x = [
          np.random.uniform(-1, 1, size=(4, in_size)).astype(
              np.float32
          ),
          np.random.uniform(-1, 1, size=(1, in_size)).astype(
              np.float32
          ),
          np.random.uniform(-1, 1, size=(3, in_size)).astype(
              np.float32
          ),
      ]
      batched_seqs = np.concatenate(seqs_x, axis=0)
      self.assertEqual(batched_seqs.shape, (8, in_size))
      sections = [4, 5]
      split = stateless_recurrent.split_batched_sequences(
          batched_seqs, sections
      )
      self.assertEqual(len(split), 3)
      np.testing.assert_allclose(split[0].array, seqs_x[0])
      np.testing.assert_allclose(split[1].array, seqs_x[1])
      np.testing.assert_allclose(split[2].array, seqs_x[2])
  
imports: import unittest
  from chainer import links
  from chainer import testing
  import numpy
  from chainerrl.links import stateless_recurrent
label_desc: Use image node_id 1 for calling the TestSplitBatchedSequences obj's underlying member method code with example usage: obj.test_array_input() without return types