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)