code: def test_erdos_renyi_graph():
      torch.manual_seed(1234)
      edge_index = erdos_renyi_graph(5, 0.2, directed=False)
      assert edge_index.tolist() == [
          [0, 1, 1, 1, 2, 4],
          [1, 0, 2, 4, 1, 1],
      ]
  
      edge_index = erdos_renyi_graph(5, 0.5, directed=True)
      assert edge_index.tolist() == [
          [1, 1, 2, 2, 3, 4, 4, 4],
          [0, 3, 0, 4, 0, 0, 1, 3],
      ]
  
imports: import numpy
  import torch
  from torch_geometric.utils import barabasi_albert_graph, erdos_renyi_graph, stochastic_blockmodel_graph
label_desc: Use image node_id 1 for calling a global function with example usage: test_erdos_renyi_graph() without return types