code: def test_benchmark_bfs_walk(self, benchmark, weighted):
      g = example_graph_random(
          n_nodes=100, n_edges=500, is_directed=True
      )
      bfw = DirectedBreadthFirstNeighbours(g)
  
      nodes = np.arange(0, 50)
      n = 5
      in_size = [5, 5]
      out_size = [5, 5]
  
      benchmark(
          lambda: bfw.run(
              nodes=nodes,
              n=n,
              in_size=in_size,
              out_size=out_size,
              weighted=weighted,
          )
      )
  
file_path: stellargraph/tests/data/test_directed_breadth_first_sampler.py
imports: import random
  import pytest
  import pandas
  import numpy
  from stellargraph.data.explorer import DirectedBreadthFirstNeighbours
  from stellargraph.core.graph import StellarDiGraph
  from ..test_utils.graphs import create_test_graph, tree_graph, example_graph_random, weighted_tree
label_desc: Use image node_id 6 for calling the TestDirectedBreadthFirstNeighbours obj's underlying member method code with example usage: obj.test_benchmark_bfs_walk(benchmark, weighted) without return types