Skip to content

Commit 4fc6792

Browse files
authored
fix: Pytests spacy (#687)
1 parent 2ba0cce commit 4fc6792

File tree

1 file changed

+45
-1
lines changed

1 file changed

+45
-1
lines changed

label_studio_ml/examples/spacy/test_api.py

Lines changed: 45 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,50 @@ def client():
1313

1414

1515
def test_predict(client):
16+
expected_response = {
17+
"results": [
18+
{
19+
"model_version": "SpacyMLBackend-v0.0.1",
20+
"result": [
21+
{
22+
"from_name": "ner_tags",
23+
"to_name": "text",
24+
"type": "labels",
25+
"value": {
26+
"end": 10,
27+
"labels": ["PERSON"],
28+
"start": 0,
29+
"text": "Katy Perry"
30+
}
31+
},
32+
{
33+
"from_name": "ner_tags",
34+
"to_name": "text",
35+
"type": "labels",
36+
"value": {
37+
"end": 23,
38+
"labels": ["ORG"],
39+
"start": 15,
40+
"text": "Pharrell"
41+
}
42+
},
43+
{
44+
"from_name": "ner_tags",
45+
"to_name": "text",
46+
"type": "labels",
47+
"value": {
48+
"end": 83,
49+
"labels": ["PRODUCT"],
50+
"start": 47,
51+
"text": "Apple Watches http://t.co/k6k3SdwShP"
52+
}
53+
}
54+
],
55+
"score": 0.0
56+
}
57+
]
58+
}
59+
1660
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}}'))
1761
assert response.status_code == 200
18-
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')
62+
assert response.json == expected_response

0 commit comments

Comments
 (0)