test.py

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"}