def checkMetricDeleted(self, uid):
engine = repository.engineFactory(config=self.__config)
with engine.begin() as conn:
with self.assertRaises(Exception) as e:
metric = repository.getMetric(conn, uid)
models = repository.getAllModels(conn)
for model in models:
self.assertNotEqual(model.uid, uid,
"Model showing up after deletion.")
with self.assertRaises(model_checkpoint_mgr.ModelNotFound):
model_checkpoint_mgr.ModelCheckpointMgr().loadModelDefinition(uid)
def checkMetricDeleted(self, uid):
engine = repository.engineFactory(config=self.__config)
with engine.begin() as conn:
with self.assertRaises(Exception) as e:
metric = repository.getMetric(conn, uid)
models = repository.getAllModels(conn)
for model in models:
self.assertNotEqual(model.uid, uid,
"Model showing up after deletion.")
with self.assertRaises(model_checkpoint_mgr.ModelNotFound):
model_checkpoint_mgr.ModelCheckpointMgr().loadModelDefinition(uid)
def checkModelDeleted(self, uid):
"""Check that the model has been deleted"""
engine = repository.engineFactory(config=self.__config)
with engine.begin() as conn:
try:
metric = repository.getMetric(conn, uid)
raise Exception("Metric not deleted as expected")
except app_exceptions.ObjectNotFoundError:
pass
models = repository.getAllModels(conn)
for model in models:
self.assertNotEqual(model.uid, uid,
"Model showing up after deletion.")
with self.assertRaises(model_checkpoint_mgr.ModelNotFound):
model_checkpoint_mgr.ModelCheckpointMgr().loadModelDefinition(uid)
def checkModelDeleted(self, uid):
"""Check that the model has been deleted"""
engine = repository.engineFactory(config=self.__config)
with engine.begin() as conn:
try:
metric = repository.getMetric(conn, uid)
raise Exception("Metric not deleted as expected")
except app_exceptions.ObjectNotFoundError:
pass
models = repository.getAllModels(conn)
for model in models:
self.assertNotEqual(model.uid, uid,
"Model showing up after deletion.")
with self.assertRaises(model_checkpoint_mgr.ModelNotFound):
model_checkpoint_mgr.ModelCheckpointMgr().loadModelDefinition(uid)
