Spaces:
Runtime error
Runtime error
File size: 1,379 Bytes
dde1233 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
from fastapi.testclient import TestClient
from main import app
from main import TextInput
client = TestClient(app)
def test_positive_sentiment():
response = client.post("/analyze", json={"text": "Profits are up by 5 million this year."})
assert response.status_code == 200
# Ensure that the response is parsed as a list of dictionaries
data_list = response.json()
assert isinstance(data_list, list)
# Check if the first item in the list has a "label" key
assert data_list[0]["label"] == "positive"
def test_negative_sentiment():
response = client.post("/analyze", json={"text": "Operating profit totaled EUR 9.4 mn , down from EUR 11.7 mn in 2004 ."})
assert response.status_code == 200
# Ensure that the response is parsed as a list of dictionaries
data_list = response.json()
assert isinstance(data_list, list)
# Check if the first item in the list has a "label" key
assert data_list[0]["label"] == "negative"
def test_neutral_sentiment():
response = client.post("/analyze", json={"text": "The profits are same as last year"})
assert response.status_code == 200
# Ensure that the response is parsed as a list of dictionaries
data_list = response.json()
assert isinstance(data_list, list)
# Check if the first item in the list has a "label" key
assert data_list[0]["label"] == "neutral"
|