code: def test_dist_tensor(g):
      first_type = g.ntypes[0]
      data_shape = (g.num_nodes(first_type), 2)
      dist_tensor_test_sanity(data_shape)
      dist_tensor_test_sanity(data_shape, name="DistTensorSanity")
      dist_tensor_test_destroy_recreate(
          data_shape, name="DistTensorRecreate"
      )
      dist_tensor_test_persistent(data_shape)
  
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 16 for calling a global function with example usage: test_dist_tensor(g) without return types