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)