code: def test_sqlite_adapter_get_batch_generator(sqlite_adapter):
      batch_gen = sqlite_adapter._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_sqlite_adapter.py
imports: from ampligraph.datasets.sqlite_adapter import SQLiteAdapter
  from ampligraph.datasets.source_identifier import DataSourceIdentifier
  import pytest
  import numpy
  import pandas
label_desc: Use image node_id 2 for calling a global function with example usage: test_sqlite_adapter_get_batch_generator(sqlite_adapter) without return types