code: def assert_correct_eids(edge_df, sample_edge_id_df):
      # We test that all src, dst correspond to the correct
      # eids in the sample_edge_id_df
      # we do this by ensuring that the inner merge to edge_df
      # remains the same as sample_edge_id_df
      # if they don't correspond correctly
      # the inner merge would fail
      sample_edge_id_df = sample_edge_id_df.sort_values(by="_EDGE_ID_")
      sample_edge_id_df = sample_edge_id_df.reset_index(drop=True)
  
      sample_merged_df = sample_edge_id_df.merge(edge_df, how="inner")
      sample_merged_df = sample_merged_df.sort_values(by="_EDGE_ID_")
      sample_merged_df = sample_merged_df.reset_index(drop=True)
      assert sample_merged_df.equals(sample_edge_id_df)
  
file_path: cugraph/python/cugraph/cugraph/tests/gnn/test_dgl_uniform_sampler.py
imports: import pytest
  import pandas
  import numpy
  import cudf
  import cupy
  from cugraph.gnn.dgl_extensions.dgl_uniform_sampler import DGLUniformSampler
label_desc: Use image node_id 1 for calling a global function with example usage: assert_correct_eids(edge_df, sample_edge_id_df) without return types