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))
