Create test.py

test.py

@@ -0,0 +1,44 @@
+from fastapi.testclient import TestClient
+from main import app
+
+client = TestClient(app)
+
+def test_positive_sentiment():
+    response = client.post("/analyze/", json={"text": "I'm thrilled with the results of this service!"})
+    assert response.status_code == 200
+    assert response.json() == {"sentiment": "joy"}
+
+def test_negative_sentiment():
+    response = client.post("/analyze/", json={"text": "What a disappointment! It didn't meet my expectations at all."})
+    assert response.status_code == 200
+    assert response.json() == {"sentiment": "anger"}
+
+def test_neutral_sentiment():
+    response = client.post("/analyze/", json={"text": "The product does exactly what it promises. Nothing more, nothing less."})
+    assert response.status_code == 200
+    assert response.json() == {"sentiment": "neutral"}
+
+def test_anger_sentiment():
+    response = client.post("/analyze/", json={"text": "This app is frustrating and annoying to use!"})
+    assert response.status_code == 200
+    assert response.json() == {"sentiment": "anger"}
+
+def test_disgust_sentiment():
+    response = client.post("/analyze/", json={"text": "The taste of this food is disgusting!"})
+    assert response.status_code == 200
+    assert response.json() == {"sentiment": "disgust"}
+
+def test_fear_sentiment():
+    response = client.post("/analyze/", json={"text": "I'm worried that this might be unsafe."})
+    assert response.status_code == 200
+    assert response.json() == {"sentiment": "fear"}
+
+def test_sadness_sentiment():
+    response = client.post("/analyze/", json={"text": "It's sad to see it perform so poorly."})
+    assert response.status_code == 200
+    assert response.json() == {"sentiment": "sadness"}
+
+def test_surprise_sentiment():
+    response = client.post("/analyze/", json={"text": "Wow, I didn't expect it to be this good!"})
+    assert response.status_code == 200
+    assert response.json() == {"sentiment": "surprise"}