ADK09's picture
Initial Commit
5c50bbf
raw
history blame contribute delete
676 Bytes
from fastapi.testclient import TestClient
from app import app
client = TestClient(app)
def test_home():
response = client.get("/")
assert response.status_code == 200
assert response.json() == {"message": "Speak your mind emotion API is running"}
def test_predict():
test_input = {"text": "I feel fantastic today!"}
response = client.post("/classify-emotion", json=test_input)
assert response.status_code == 200
assert "predicted_emotion" in response.json()
assert isinstance(response.json()["predicted_emotion"], str)
if __name__ == "__main__":
test_home()
test_predict()
print("All tests passed successfully!")