def predict(path='/predict?sepal length=1.0&sepal width=1.1&' 'petal length=0.777&petal width=5'): from palladium.server import app from palladium.server import predict with app.test_request_context(path): response = predict() assert response.status_code == 200
def test_unknown_exception(self, predict, flask_app): model_persister = Mock() model_persister.read.side_effect = KeyError('model') with flask_app.test_request_context(): resp = predict(model_persister, Mock()) resp_data = json.loads(resp.get_data(as_text=True)) assert resp.status_code == 500 assert resp_data == { "status": "ERROR", "error_code": -1, "error_message": "KeyError: 'model'", }
def predict(path='/predict?sepal length=1.0&sepal width=1.1&' 'petal length=0.777&petal width=5', predict_service_name='predict_service'): from palladium.server import app from palladium.server import predict from palladium.util import get_config model_persister = get_config()['model_persister'] predict_service = get_config()[predict_service_name] with app.test_request_context(path): response = predict(model_persister, predict_service) assert response.status_code == 200