from fastapi.testclient import TestClient | |
from main import app | |
client = TestClient(app) | |
def test_positive_sentiment(): | |
response = client.post("/predict", json={"text": "I love this product! It's amazing."}) | |
assert response.status_code == 200 | |
assert response.json() == {"sentiment": "positive"} | |
def test_negative_sentiment(): | |
response = client.post("/predict", json={"text": "This product is terrible. I regret buying it."}) | |
assert response.status_code == 200 | |
assert response.json() == {"sentiment": "negative"} | |
def test_neutral_sentiment(): | |
response = client.post("/predict", json={"text": "This product is okay. It meets my expectations."}) | |
assert response.status_code == 200 | |
assert response.json() == {"sentiment": "neutral"} | |