code: def test_edge_prediction_sampler(idtype):
      g = create_test_graph(idtype)
      sampler = NeighborSampler([10, 10])
      sampler = as_edge_prediction_sampler(
          sampler, negative_sampler=negative_sampler.Uniform(1)
      )
  
      seeds = F.copy_to(F.arange(0, 2, dtype=idtype), ctx=F.ctx())
      # just a smoke test to make sure we don't fail internal assertions
      result = sampler.sample(g, {"follows": seeds})
  
imports: import unittest
  import backend
  import dgl
  from dgl.dataloading import as_edge_prediction_sampler, negative_sampler, NeighborSampler
  from utils import parametrize_idtype
label_desc: Use image node_id 2 for calling a global function with example usage: test_edge_prediction_sampler(idtype) without return types