code: def test_to_dense_adj(self):
      edge_index = tf.constant([[0, 0, 1, 2, 3, 4], [0, 1, 0, 3, 4, 2]])
      batch = tf.constant([0, 0, 1, 1, 1])
  
      adj = to_dense_adj(edge_index, batch)
  
      init = tf.global_variables_initializer()
      with tf.Session() as sess:
          sess.run(init)
          ret = sess.run(adj)
  
          assert ret.shape == (2, 3, 3)
          assert ret[0].tolist() == [[1, 1, 0], [1, 0, 0], [0, 0, 0]]
          assert ret[1].tolist() == [[0, 1, 0], [0, 0, 1], [1, 0, 0]]
  
imports: from tensorflow.python.platform import test
  from tf_euler.python.utils import to_dense_adj
  import tensorflow
label_desc: Use image node_id 1 for calling the ToDenseAdjTest obj's underlying member method code with example usage: obj.test_to_dense_adj() without return types