code: def test_ba_graph():
      graph_generator = BAGraph(num_nodes=300, num_edges=5)
      assert (
          str(graph_generator) == "BAGraph(num_nodes=300, num_edges=5)"
      )
  
      data = graph_generator()
      assert len(data) == 2
      assert data.num_nodes == 300
      assert data.num_edges <= 2 * 300 * 5
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_ba_graph() without return types