def test_copy_graph(self):
"""
test that a copy of a graph has the same number of nodes and edges and that the primary keys have been changed
and that the actual node references are different
"""
root = models.Node.objects.get(pk=self.HERITAGE_RESOURCE_FIXTURE)
graph = Graph(root)
graph_copy = graph.copy()
self.assertEqual(len(graph.nodes), len(graph_copy.nodes))
self.assertEqual(len(graph.edges), len(graph_copy.edges))
self.assertEqual(len(graph.nodegroups), len(graph_copy.nodegroups))
def findNodeByName(graph, name):
for key, node in graph.nodes.iteritems():
if node.name == name:
return node
return None
for key, node in graph.nodes.iteritems():
node_copy = findNodeByName(graph_copy, node.name)
self.assertIsNotNone(node_copy)
self.assertNotEqual(node.pk, node_copy.pk)
self.assertNotEqual(id(node), id(node_copy))
self.assertEqual(node.is_collector(), node_copy.is_collector())
if node.nodegroup != None:
self.assertNotEqual(node.nodegroup, node_copy.nodegroup)
for key, newedge in graph_copy.edges.iteritems():
self.assertIsNotNone(graph_copy.nodes[newedge.domainnode_id])
self.assertIsNotNone(graph_copy.nodes[newedge.rangenode_id])
self.assertEqual(newedge.domainnode, graph_copy.nodes[newedge.domainnode.pk])
self.assertEqual(newedge.rangenode, graph_copy.nodes[newedge.rangenode.pk])
with self.assertRaises(KeyError):
graph.edges[newedge.pk]
