code: def test_rmat_edge_ids(scale, include_edge_ids, scramble_vertex_ids):
      """
      Verifies that the edge ids returned by rmat() are valid.
  
      """
      num_edges = (2**scale) * 4
      create_using = None  # Returns the edgelist from RMAT
      df = _call_rmat(
          scale,
          num_edges,
          create_using,
          scramble_vertex_ids=scramble_vertex_ids,
          include_edge_ids=include_edge_ids,
      )
  
      if include_edge_ids:
          assert "edge_id" in df.columns
          df["index"] = df.index
          edge_id_err = df.query("index != edge_id")
          assert len(edge_id_err) == 0
  
      else:
          assert len(df.columns) == 2
  
file_path: cugraph/python/cugraph/cugraph/tests/generators/test_rmat.py
imports: import pytest
  import cudf
  import cugraph
  from cugraph.generators import rmat
  from cupy.sparse import coo_matrix, triu, tril
  import numpy
  import cupy
label_desc: Use image node_id 3 for calling a global function with example usage: test_rmat_edge_ids(scale, include_edge_ids, scramble_vertex_ids) without return types