def test_clone(self): G, x, y, edge_x, edge_y, edge_index, graph_x, graph_y = ( simple_networkx_graph() ) dg = Graph( node_feature=x, node_label=y, edge_index=edge_index, edge_feature=edge_x, edge_label=edge_y, graph_feature=graph_x, graph_label=graph_y, directed=True ) dg1 = dg.clone() self.assertEqual(dg.num_nodes, dg1.num_nodes) self.assertEqual(dg.num_edges, dg1.num_edges) self.assertEqual(dg.num_node_features, dg1.num_node_features) self.assertEqual(dg.num_edge_features, dg1.num_edge_features) self.assertEqual(dg.num_node_labels, dg1.num_node_labels) self.assertEqual(dg.num_edge_labels, dg1.num_edge_labels) self.assertTrue(not id(dg.edge_index) == id(dg1.edge_index)) self.assertTrue(tuple(dg.keys) == tuple(dg1.keys))
def test_clone(self): G, x, y, edge_x, edge_y, edge_index, graph_x, graph_y = \ simple_networkx_graph() Graph.add_edge_attr(G, "edge_feature", edge_x) Graph.add_edge_attr(G, "edge_label", edge_y) Graph.add_node_attr(G, "node_feature", x) Graph.add_node_attr(G, "node_label", y) Graph.add_graph_attr(G, "graph_feature", graph_x) Graph.add_graph_attr(G, "graph_label", graph_y) dg = Graph(G) dg1 = dg.clone() self.assertEqual(dg.num_nodes, dg1.num_nodes) self.assertEqual(dg.num_edges, dg1.num_edges) self.assertEqual(dg.num_node_features, dg1.num_node_features) self.assertEqual(dg.num_edge_features, dg1.num_edge_features) self.assertEqual(dg.num_node_labels, dg1.num_node_labels) self.assertEqual(dg.num_edge_labels, dg1.num_edge_labels) self.assertTrue(not id(dg.G) == id(dg1.G)) self.assertTrue(tuple(dg.keys) == tuple(dg1.keys))