code: def test_get_edge_index(graph, edge_index_type, dask_client):
      F, G, N = graph
      if "torch" in edge_index_type:
          if edge_index_type == "torch-cpu":
              device = "cpu"
          else:
              device = "cuda"
          for et in list(G.keys()):
              G[et][0] = torch.as_tensor(G[et][0], device=device)
              G[et][1] = torch.as_tensor(G[et][1], device=device)
      elif edge_index_type == "cudf":
          for et in list(G.keys()):
              G[et][0] = cudf.Series(G[et][0])
              G[et][1] = cudf.Series(G[et][1])
      elif edge_index_type == "dask-cudf":
          for et in list(G.keys()):
              G[et][0] = dask_cudf.from_cudf(
                  cudf.Series(G[et][0]), npartitions=1
              )
              G[et][1] = dask_cudf.from_cudf(
                  cudf.Series(G[et][1]), npartitions=1
              )
  
      cugraph_store = CuGraphStore(F, G, N, order="CSC", multi_gpu=True)
  
      for pyg_can_edge_type in G:
          src, dst = cugraph_store.get_edge_index(
              edge_type=pyg_can_edge_type, layout="coo", is_sorted=False
          )
  
          if edge_index_type == "cudf":
              assert (
                  G[pyg_can_edge_type][0].values_host.tolist()
                  == src.tolist()
              )
              assert (
                  G[pyg_can_edge_type][1].values_host.tolist()
                  == dst.tolist()
              )
          elif edge_index_type == "dask-cudf":
              assert (
                  G[pyg_can_edge_type][0].compute().values_host.tolist()
                  == src.tolist()
              )
              assert (
                  G[pyg_can_edge_type][1].compute().values_host.tolist()
                  == dst.tolist()
              )
          else:
              assert G[pyg_can_edge_type][0].tolist() == src.tolist()
              assert G[pyg_can_edge_type][1].tolist() == dst.tolist()
  
imports: import cugraph
  from cugraph_pyg.data.cugraph_store import CuGraphTensorAttr, CuGraphEdgeAttr, EdgeLayout
  from cugraph_pyg.data import CuGraphStore
  import cudf
  import dask_cudf
  import cupy
  import numpy
  from cugraph.utilities.utils import import_optional, MissingModule
  import pytest
label_desc: Use image node_id 5 for calling a global function with example usage: test_get_edge_index(graph, edge_index_type, dask_client) without return types