code: def test_dist_dataloader(g):
      orig_nid_map = dict()
      dtype = g.edges[g.etypes[0]].data["edge_u"].dtype
      for ntype in g.ntypes:
          orig_nid = F.tensor(
              np.load(graph_path + f"/orig_nid_array_{ntype}.npy"),
              dtype,
          )
          orig_nid_map[ntype] = orig_nid
  
      orig_uv_map = dict()
      for etype in g.etypes:
          orig_uv_map[etype] = (
              g.edges[etype].data["edge_u"],
              g.edges[etype].data["edge_v"],
          )
  
      batch_size_l = [64]
      drop_last_l = [False, True]
      num_workers_l = [0, 4]
      shuffle_l = [False, True]
  
      for batch_size, drop_last, shuffle, num_workers in product(
          batch_size_l, drop_last_l, shuffle_l, num_workers_l
      ):
          if len(g.ntypes) == 1 and num_workers == 0:
              distdataloader_test(g, batch_size, drop_last, shuffle)
          distnodedataloader_test(
              g,
              batch_size,
              drop_last,
              shuffle,
              num_workers,
              orig_nid_map,
              orig_uv_map,
          )
          # No negssampling
          distedgedataloader_test(
              g,
              batch_size,
              drop_last,
              shuffle,
              num_workers,
              orig_nid_map,
              orig_uv_map,
              num_negs=0,
          )
          # negsampling 15
          distedgedataloader_test(
              g,
              batch_size,
              drop_last,
              shuffle,
              num_workers,
              orig_nid_map,
              orig_uv_map,
              num_negs=15,
          )
  
      multi_distdataloader_test(g, dgl.dataloading.DistNodeDataLoader)
      multi_distdataloader_test(g, dgl.dataloading.DistEdgeDataLoader)
  
imports: import json
  import os
  from itertools import product
  import dgl
  import dgl.backend
  import numpy
  from dgl.distributed import edge_split, load_partition_book, node_split
label_desc: Use image node_id 26 for calling a global function with example usage: test_dist_dataloader(g) without return types