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, [])
