def testModelCommandResultSerializableStateWithNonZeroStatus(self): commandID = 1 method = "testMethod" status = 1 errorMessage = "something bad happened" commandResult = ModelCommandResult(commandID=commandID, method=method, status=status, errorMessage=errorMessage) self.assertEqual(commandResult.commandID, commandID) self.assertEqual(commandResult.method, method) self.assertEqual(commandResult.status, status) self.assertEqual(commandResult.errorMessage, errorMessage) self.assertIsNone(commandResult.args) self.assertIn("ModelCommandResult<", str(commandResult)) self.assertIn("ModelCommandResult<", repr(commandResult)) commandResult2 = _ModelRequestResultBase.__createFromState__( commandResult.__getstate__()) self.assertEqual(commandResult2.commandID, commandID) self.assertEqual(commandResult2.method, method) self.assertEqual(commandResult2.status, status) self.assertEqual(commandResult2.errorMessage, errorMessage) self.assertIsNone(commandResult2.args) self.assertIn("ModelCommandResult<", str(commandResult2)) self.assertIn("ModelCommandResult<", repr(commandResult2))
def testModelInputRowConstructorFromSerializableState(self): rowID = 1 data = [1, 2, datetime.datetime.utcnow()] inputRow = ModelInputRow(rowID=rowID, data=copy.copy(data)) self.assertEqual(inputRow.rowID, rowID) self.assertEqual(inputRow.data, data) self.assertIn("ModelInputRow<", str(inputRow)) self.assertIn("ModelInputRow<", repr(inputRow)) inputRow2 = _ModelRequestResultBase.__createFromState__( inputRow.__getstate__()) self.assertEqual(inputRow2.rowID, rowID) self.assertEqual(inputRow2.data, data) self.assertIn("ModelInputRow<", str(inputRow2)) self.assertIn("ModelInputRow<", repr(inputRow2))
def testModelCommandSerializableState(self): commandID = 1 method = "defineModel" args = {'key1': 4098, 'key2': 4139} command = ModelCommand(commandID=commandID, method=method, args=copy.copy(args)) self.assertEqual(command.commandID, commandID) self.assertEqual(command.method, method) self.assertEqual(command.args, args) command2 = _ModelRequestResultBase.__createFromState__( command.__getstate__()) self.assertEqual(command2.commandID, commandID) self.assertEqual(command2.method, method) self.assertEqual(command2.args, args) self.assertIn("ModelCommand<", str(command2)) self.assertIn("ModelCommand<", repr(command2))
def testModelInferenceResultSerializableStateWithErrorMessage(self): rowID = 1 status = 1 errorMessage = "error" inferenceResult = ModelInferenceResult(rowID=rowID, status=status, errorMessage=errorMessage) self.assertEqual(inferenceResult.rowID, rowID) self.assertEqual(inferenceResult.status, status) self.assertEqual(inferenceResult.errorMessage, errorMessage) self.assertIsNone(inferenceResult.anomalyScore) self.assertIn("ModelInferenceResult<", str(inferenceResult)) self.assertIn("ModelInferenceResult<", repr(inferenceResult)) inferenceResult2 = _ModelRequestResultBase.__createFromState__( inferenceResult.__getstate__()) self.assertEqual(inferenceResult2.rowID, rowID) self.assertEqual(inferenceResult2.status, status) self.assertEqual(inferenceResult2.errorMessage, errorMessage) self.assertIsNone(inferenceResult2.anomalyScore) self.assertIn("ModelInferenceResult<", str(inferenceResult2)) self.assertIn("ModelInferenceResult<", repr(inferenceResult2))
def _implModelCommandResultSerializableStateWithZeroStatus(self, args): commandID = 1 method = "testMethod" status = 0 commandResult = ModelCommandResult(commandID=commandID, method=method, status=status, args=copy.copy(args)) self.assertEqual(commandResult.commandID, commandID) self.assertEqual(commandResult.method, method) self.assertEqual(commandResult.status, status) self.assertEqual(commandResult.args, args) self.assertIsNone(commandResult.errorMessage) commandResult2 = _ModelRequestResultBase.__createFromState__( commandResult.__getstate__()) self.assertEqual(commandResult2.commandID, commandID) self.assertEqual(commandResult2.method, method) self.assertEqual(commandResult2.status, status) self.assertEqual(commandResult2.args, args) self.assertIsNone(commandResult2.errorMessage) self.assertIn("ModelCommandResult<", str(commandResult2)) self.assertIn("ModelCommandResult<", repr(commandResult2))