code: def _test_batch_mode(model, edges=False, **kwargs):
      dataset = TestDataset(
          [
              _get_graph(
                  n_nodes=n_nodes,
                  n_features=n_node_features,
                  n_edge_features=n_edge_features if edges else None,
              )
              for _ in range(batch_size)
          ]
      )
  
      loader = loaders.BatchLoader(
          dataset, epochs=1, batch_size=batch_size
      )
      inputs = loader.__next__()
  
      model_instance = model(**kwargs)
      output = model_instance(inputs)
  
file_path: spektral/tests/test_models/core.py
imports: import numpy
  import scipy.sparse
  import tensorflow
  from spektral.data import Dataset, Graph, loaders
label_desc: Use image node_id 4 for calling a global function with example usage: _test_batch_mode(model, edges) without return types