code: def test_flow_batch_size(barbell, num_powers):
      batch_size = 2
      generator = AdjacencyPowerGenerator(
          barbell, num_powers=num_powers
      )
      for x, y in generator.flow(batch_size=batch_size).take(1):
          assert x[0].shape == (batch_size,)
          assert y.shape == (batch_size, 1, len(barbell.nodes()))
          assert x[1].shape == (
              batch_size,
              num_powers,
              len(barbell.nodes()),
          )
  
file_path: stellargraph/tests/mapper/test_adjacency_generators.py
imports: from stellargraph.core.utils import normalize_adj
  from stellargraph.mapper.adjacency_generators import AdjacencyPowerGenerator
  from ..test_utils.graphs import barbell
  import tensorflow
  import numpy
  import pytest
label_desc: Use image node_id 6 for calling a global function with example usage: test_flow_batch_size(barbell, num_powers) without return types