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_03_loadModelForError(self):
logging.info("Test Case : Load a model into memory. (2)")
filePath = os.path.abspath('testUseCase/supportdata/errorPmml.pmml')
result = Scoring.loadModelfile(filePath)
self.assertEqual(result.status_code, 500)
self.assertEqual(
json.loads(result.__dict__['_container'][0])['keytoModel'], None)
self.assertEqual(
json.loads(result.__dict__['_container'][0])['message'],
'Model loading failed, please contact Admin')
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 test_02_loadModelForCorrectness(self):
logging.info("Test Case : Load a model into memory. (1)")
filePath = os.path.abspath(
'testUseCase/supportdata/NewTrialModel.pmml')
result = Scoring.loadModelfile(filePath)
self.assertEqual(result.status_code, 200)
self.assertEqual(
json.loads(result.__dict__['_container'][0])['keytoModel'],
'NewTrialModel')
self.assertEqual(
json.loads(result.__dict__['_container'][0])['message'],
'Model loaded successfully')
result = Scoring.getListOfModelinMemory()
self.assertEqual(len(json.loads(result.__dict__['_container'][0])), 1)
logging.info("PASSED")
def test_02_loadModelForCorrectness(self):
logging.info("Test Case : Load a model into memory. (1)")
filePath = os.path.abspath('testUseCase/supportData2/mpgSKModel.pmml')
# result = Scoring.loadModelfile(filePath)
result = NewModelOperations().loadExecutionModel(filePath)
self.assertEqual(result.status_code, 200)
self.assertEqual(
json.loads(result.__dict__['_container'][0])['keytoModel'],
'mpgSKModel')
self.assertEqual(
json.loads(result.__dict__['_container'][0])['message'],
'Model Loaded Successfully')
result = Scoring.getListOfModelinMemory()
# print ('>>>>>>','result >>>>>>>>>',result.__dict__['_container'])
self.assertEqual(len(json.loads(result.__dict__['_container'][0])), 1)
logging.info("PASSED")
def get(self, requests):
return Scoring.getListOfModelinMemory()
def test_01_listOfLoadedModels(self):
logging.info("Test Case : Get list of loaded models.")
result = Scoring.getListOfModelinMemory()
self.assertEqual(len(json.loads(result.__dict__['_container'][0])), 0)
logging.info("PASSED")