Example #1
0
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
Example #2
0
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
Example #3
0
    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'",
        }
Example #4
0
    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'",
            }
Example #5
0
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