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)