def testRemoveAll(self): """ Test removeAll """ checkpointMgr = ModelCheckpointMgr() # Should be empty at first ids = checkpointMgr.getModelIDs() self.assertSequenceEqual(ids, []) # Create some checkpoints using meta info expModelIDs = [uuid.uuid1().hex, uuid.uuid1().hex] expModelIDs.sort() for modelID in expModelIDs: checkpointMgr.define(modelID, definition={'a':1}) ids = checkpointMgr.getModelIDs() self.assertItemsEqual(ids, expModelIDs) # Delete checkpoint store ModelCheckpointMgr.removeAll() ids = checkpointMgr.getModelIDs() self.assertSequenceEqual(ids, [])
def testRemoveAll(self): """ Test removeAll """ checkpointMgr = ModelCheckpointMgr() # Should be empty at first ids = checkpointMgr.getModelIDs() self.assertSequenceEqual(ids, []) # Create some checkpoints using meta info expModelIDs = [uuid.uuid1().hex, uuid.uuid1().hex] expModelIDs.sort() for modelID in expModelIDs: checkpointMgr.define(modelID, definition={'a':1}) ids = checkpointMgr.getModelIDs() self.assertItemsEqual(ids, expModelIDs) # Delete checkpoint store ModelCheckpointMgr.removeAll() ids = checkpointMgr.getModelIDs() self.assertSequenceEqual(ids, [])
def testRemoveAndGetModelIDs(self): """ Test getModelIDs and remove methods """ checkpointMgr = ModelCheckpointMgr() # Should be empty at first ids = checkpointMgr.getModelIDs() self.assertListEqual(ids, []) # Create some checkpoints using meta info expModelIDs = [uuid.uuid1().hex, uuid.uuid1().hex] expModelIDs.sort() for modelID in expModelIDs: checkpointMgr.define(modelID, definition={'a':1}) ids = checkpointMgr.getModelIDs() ids.sort() self.assertListEqual(ids, expModelIDs) # Delete one of them checkpointMgr.remove(expModelIDs[0]) expModelIDs.remove(expModelIDs[0]) ids = checkpointMgr.getModelIDs() ids.sort() self.assertListEqual(ids, expModelIDs) # Delete all of them for modelID in expModelIDs: checkpointMgr.remove(modelID) ids = checkpointMgr.getModelIDs() self.assertListEqual(ids, []) # If we try and delete a non-existing model, should get an exception self.assertRaises(ModelNotFound, checkpointMgr.remove, "IDx")
def testRemoveAndGetModelIDs(self): """ Test getModelIDs and remove methods """ checkpointMgr = ModelCheckpointMgr() # Should be empty at first ids = checkpointMgr.getModelIDs() self.assertListEqual(ids, []) # Create some checkpoints using meta info expModelIDs = [uuid.uuid1().hex, uuid.uuid1().hex] expModelIDs.sort() for modelID in expModelIDs: checkpointMgr.define(modelID, definition={'a':1}) ids = checkpointMgr.getModelIDs() ids.sort() self.assertListEqual(ids, expModelIDs) # Delete one of them checkpointMgr.remove(expModelIDs[0]) expModelIDs.remove(expModelIDs[0]) ids = checkpointMgr.getModelIDs() ids.sort() self.assertListEqual(ids, expModelIDs) # Delete all of them for modelID in expModelIDs: checkpointMgr.remove(modelID) ids = checkpointMgr.getModelIDs() self.assertListEqual(ids, []) # If we try and delete a non-existing model, should get an exception self.assertRaises(ModelNotFound, checkpointMgr.remove, "IDx")