code: def test_benchmark_sampledheterogeneousbreadthfirstwalk(
      self, benchmark
  ):
      g = example_graph_random(
          n_nodes=50, n_edges=250, node_types=2, edge_types=2
      )
      bfw = SampledHeterogeneousBreadthFirstWalk(g)
  
      nodes = np.arange(0, 50)
      n = 5
      n_size = [5, 5]
  
      benchmark(lambda: bfw.run(nodes=nodes, n=n, n_size=n_size))
  
file_path: stellargraph/tests/data/test_heterogeneous_breadth_first_walker.py
imports: import pandas
  import numpy
  import pytest
  from stellargraph.data.explorer import SampledHeterogeneousBreadthFirstWalk
  from stellargraph.core.graph import StellarGraph
  from ..test_utils.graphs import example_graph_random
label_desc: Use image node_id 6 for calling the TestSampledHeterogeneousBreadthFirstWalk obj's underlying member method code with example usage: obj.test_benchmark_sampledheterogeneousbreadthfirstwalk(benchmark) without return types