code: def test_to_dense_adj_with_attr2(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])
      batch_size = 2
      edge_attr = tf.constant([1, 2, 3, 4, 5, 6])
      edge_attr = tf.reshape(edge_attr, [-1, 1])
  
      adj = to_dense_adj(edge_index, batch, edge_attr)
  
      init = tf.global_variables_initializer()
      with tf.Session() as sess:
          sess.run(init)
          ret = sess.run(adj)
  
          assert ret.shape == (2, 3, 3, 1)
          assert ret[0, 0].tolist() == [[1], [2], [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 3 for calling the ToDenseAdjTest obj's underlying member method code with example usage: obj.test_to_dense_adj_with_attr2() without return types