code: def test_dist_graph(g):
      gpb_path = graph_path + "/{}.json".format(graph_name)
      with open(gpb_path) as conf_f:
          part_metadata = json.load(conf_f)
      assert "num_nodes" in part_metadata
      assert "num_edges" in part_metadata
      num_nodes = part_metadata["num_nodes"]
      num_edges = part_metadata["num_edges"]
  
      assert g.num_nodes() == num_nodes
      assert g.num_edges() == num_edges
  
      num_nodes = {ntype: g.num_nodes(ntype) for ntype in g.ntypes}
      num_edges = {etype: g.num_edges(etype) for etype in g.etypes}
  
      for key, n_nodes in num_nodes.items():
          assert g.num_nodes(key) == n_nodes
          node_split_test(g, force_even=False, ntype=key)
          node_split_test(g, force_even=True, ntype=key)
  
      for key, n_edges in num_edges.items():
          assert g.num_edges(key) == n_edges
          edge_split_test(g, force_even=False, etype=key)
          edge_split_test(g, force_even=True, etype=key)
  
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 7 for calling a global function with example usage: test_dist_graph(g) without return types