def testAddVertices(self): vList = GeneralVertexList(10) vList.setVertex(1, 2) self.assertEquals(vList.getNumVertices(), 10) self.assertEquals(vList[1], 2) vList.addVertices(5) self.assertEquals(vList.getNumVertices(), 15) vList.setVertex(11, 2) self.assertEquals(vList[1], 2) self.assertEquals(vList[1], 2)
def testSaveLoad(self): try: vList = GeneralVertexList(self.numVertices) vList.setVertex(0, "abc") vList.setVertex(1, 12) vList.setVertex(2, "num") tempDir = PathDefaults.getTempDir() fileName = tempDir + "vList" vList.save(fileName) vList2 = GeneralVertexList.load(fileName) for i in range(self.numVertices): self.assertEquals(vList.getVertex(i), vList2.getVertex(i)) except IOError as e: logging.warn(e) pass