def test_create_complete_edges_from_nodes_dynamic_number_of_nodes(self):
for graph_dict in self.graphs_dicts_in:
n_node = tf.shape(tf.constant(graph_dict["nodes"]))[0]
edges_dict = utils_tf._create_complete_edges_from_nodes_dynamic(
n_node, exclude_self_edges=False)
n_relation = n_node**2
receivers = edges_dict["receivers"].numpy()
senders = edges_dict["senders"].numpy()
n_edge = edges_dict["n_edge"].numpy()
self.assertAllEqual((n_relation,), receivers.shape)
self.assertAllEqual((n_relation,), senders.shape)
self.assertEqual(n_relation, n_edge)
def test_create_complete_edges_from_nodes_dynamic_number_of_nodes(self):
for graph_dict in self.graphs_dicts_in:
n_node = tf.shape(tf.constant(graph_dict["nodes"]))[0]
edges_dict = utils_tf._create_complete_edges_from_nodes_dynamic(
n_node, exclude_self_edges=False)
n_relation_op = n_node**2
with self.test_session() as sess:
n_relation, receivers, senders, n_edge = sess.run([
n_relation_op, edges_dict["receivers"], edges_dict["senders"],
edges_dict["n_edge"]
])
self.assertAllEqual((n_relation,), receivers.shape)
self.assertAllEqual((n_relation,), senders.shape)
self.assertEqual(n_relation, n_edge)
