code: def test_graph_data_loader_get_batch_generator(graph_data_loader):
      batch_gen = graph_data_loader.get_batch_generator()
      batch = next(batch_gen)
      assert np.shape(batch) == (
          1,
          3,
      ), "batch size is wrong, got {}".format(batch)
  
file_path: AmpliGraph/tests/ampligraph/datasets/test_graph_data_loader.py
imports: from ampligraph.datasets import GraphDataLoader, NoBackend
  from ampligraph.datasets.sqlite_adapter import SQLiteAdapter
  import pytest
  import numpy
  import pandas
label_desc: Use image node_id 3 for calling a global function with example usage: test_graph_data_loader_get_batch_generator(graph_data_loader) without return types