def test_04_unloadModelForError(self): logging.info("Test Case : Remove a loaded model from memory. (1)") modelName = 'id' result = Scoring.removeModelfromMemory(modelName) self.assertEqual(result.status_code, 500) self.assertEqual( 'message' in json.loads(result.__dict__['_container'][0]), True) self.assertEqual( json.loads(result.__dict__['_container'][0])['message'], 'Not able to locate, make sure the model was loaded') result = Scoring.getListOfModelinMemory() self.assertEqual(len(json.loads(result.__dict__['_container'][0])), 1) logging.info("PASSED")
def test_05_unloadModelForCorrectness(self): logging.info("Test Case : Remove a loaded model from memory. (2)") modelName = 'NewTrialModel' result = Scoring.removeModelfromMemory(modelName) self.assertEqual(result.status_code, 200) self.assertEqual( 'message' in json.loads(result.__dict__['_container'][0]), True) self.assertEqual( json.loads(result.__dict__['_container'][0])['message'], 'Model unloaded successfully, now it will not be available for predictions.' ) result = Scoring.getListOfModelinMemory() self.assertEqual(len(json.loads(result.__dict__['_container'][0])), 0) logging.info("PASSED")
def delete(self,requests,modelName): print('>>>>>>>>>>>>>>',modelName) return Scoring.removeModelfromMemory(modelName)