code: def test_get_entities_in_batches(data_indexer):
      for batch in data_indexer.get_entities_in_batches(batch_size=3):
          assert len(batch) == 3
          break
      for batch in data_indexer.get_entities_in_batches():
          assert len(batch) == data_indexer.get_entities_count()
  
file_path: AmpliGraph/tests/ampligraph/datasets/test_data_indexer.py
imports: from ampligraph.datasets.data_indexer import DataIndexer
  import numpy
  import pytest
label_desc: Use image node_id 6 for calling a global function with example usage: test_get_entities_in_batches(data_indexer) without return types