def testDifferentGraph(self): expectedGraph, expectedNodeMap = self.createUngroupedGraph2() badGraph, badNodeMap = self.createUngroupedGraph1() try: GraphModule.assertGraphAutomorphism(expectedGraph, badGraph, lambda x: x.nodes()[0]) raise ErrorModule.InvalidValueError("expected to fail") except AssertionError: pass return
def _testGroupNodes(self): actualGraph, actualNodeMap = self.createUngroupedGraph1() expectedGraph, expectedNodeMap = self.createGroupedGraph1() nodesToGroup = [] for id in [4,5,6,7,8]: node = actualNodeMap[id] nodesToGroup.append(node) logging.debug("appended node %s for id %s" % (node.id(), id)) group = actualGraph.groupNodes(nodesToGroup) group.id(11) GraphModule.assertGraphAutomorphism(expectedGraph, actualGraph, lambda x: x.nodes()[-1]) pass
def _testUngroupNode(self): expectedGraph, expectedNodeMap = self.createUngroupedGraph2() actualGraph, actualNodeMap = self.createGroupedGraph1() badGraph, badNodeMap = self.createUngroupedGraph1() actualGraph.ungroupNode(actualNodeMap[11]) logging.debug("testUngroupNode expectedNodes >> %s" % expectedGraph.nodes()) logging.debug("testUngroupNode actualNodes >> %s" % actualGraph.nodes()) logging.debug("testUngroupNode expectedEdges >> %s" % expectedGraph.edges()) logging.debug("testUngroupNode actualNodes >> %s" % actualGraph.edges()) GraphModule.assertGraphAutomorphism(expectedGraph, actualGraph, lambda x: x.nodes()[0]) pass