code: def test_to_batch():
      # TODO test e_list
      x = to_batch(x_list=x_list)
      a = to_batch(a_list=a_list)
      x, a = to_batch(x_list, a_list)
      assert x.ndim == 3
      assert a.ndim == 3
      assert x.shape[0] == a.shape[0]
      assert x.shape[1] == a.shape[1] == a.shape[2]
  
imports: import numpy
  import scipy.sparse
  from spektral.data import Dataset, Graph
  from spektral.data.utils import batch_generator, to_batch, to_disjoint
label_desc: Use image node_id 1 for calling a global function with example usage: test_to_batch() without return types