code: def test_renumber_vertices_multi_edge_multi_vertex(
      multi_edge_multi_vertex_graph_1, dask_client
  ):
      F, G, N = multi_edge_multi_vertex_graph_1
      cugraph_store = CuGraphStore(F, G, N, multi_gpu=True)
  
      nodes_of_interest = torch.as_tensor(
          cupy.random.randint(0, sum(N.values()), 3), device="cuda"
      ).unique()
  
      index = cugraph_store._get_vertex_groups_from_sample(
          nodes_of_interest
      )
  
      black_nodes = nodes_of_interest[nodes_of_interest <= 1]
      brown_nodes = nodes_of_interest[nodes_of_interest > 1] - 2
  
      if len(black_nodes) > 0:
          assert index["black"].tolist() == sorted(black_nodes.tolist())
      if len(brown_nodes) > 0:
          assert index["brown"].tolist() == sorted(brown_nodes.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 9 for calling a global function with example usage: test_renumber_vertices_multi_edge_multi_vertex(multi_edge_multi_vertex_graph_1, dask_client) without return types