test_api.py

import pytest
import json
from model import SpacyMLBackend


@pytest.fixture
def client():
    from _wsgi import init_app
    app = init_app(model_class=SpacyMLBackend)
    app.config['TESTING'] = True
    with app.test_client() as client:
        yield client


def test_predict(client):
    response = client.post('/predict', json=json.loads('{"tasks": [{"id": 1428, "data": {"text": "Katy Perry and Pharrell have the same taste in Apple Watches http://t.co/k6k3SdwShP"}, "meta": {}, "created_at": "2024-03-20T20:50:22.955810Z", "updated_at": "2024-03-20T20:50:22.955819Z", "is_labeled": false, "overlap": 1, "inner_id": 18, "total_annotations": 0, "cancelled_annotations": 0, "total_predictions": 0, "comment_count": 0, "unresolved_comment_count": 0, "last_comment_updated_at": null, "project": 6, "updated_by": null, "file_upload": 8, "comment_authors": [], "annotations": [], "predictions": []}], "project": "6.1710967805", "label_config": "<View>\\n  <Labels name=\\"ner_tags\\" toName=\\"text\\">\\n    <Label value=\\"label_1\\" background=\\"#FFA39E\\"/>\\n    <Label value=\\"label_2\\" background=\\"#D4380D\\"/>\\n    <Label value=\\"label_3\\" background=\\"#FFC069\\"/>\\n  </Labels>\\n  <Text name=\\"text\\" value=\\"$text\\"/>\\n</View>", "params": {"login": null, "password": null, "context": null}}'))
    assert response.status_code == 200
    assert response.json == json.loads('{"results":[{"model_version":"SpacyMLBackend-v0.0.1","result":[{"from_name":"ner_tags","to_name":"text","type":"labels","value":{"end":10,"labels":["PERSON"],"start":0,"text":"Katy Perry"}},{"from_name":"ner_tags","to_name":"text","type":"labels","value":{"end":23,"labels":["ORG"],"start":15,"text":"Pharrell"}},{"from_name":"ner_tags","to_name":"text","type":"labels","value":{"end":60,"labels":["ORG"],"start":47,"text":"Apple Watches"}}],"score":0.0}]}\n')