from fastapi.testclient import TestClient from app import app from app import TextInput from fastapi.encoders import jsonable_encoder client = TestClient(app) # Test the welcome endpoint def test_welcome(): # Test the welcome endpoint response = client.get("/") assert response.status_code == 200 assert response.json() == "Welcome to our Text Classification API" # Test the sentiment analysis endpoint for positive sentiment def test_positive_sentiment(): with client: # Define the request payload # Initialize payload as a TextInput object payload = TextInput(text="I love this product! It's amazing!") # Convert TextInput object to JSON-serializable dictionary payload_dict = jsonable_encoder(payload) # Send a POST request to the sentiment analysis endpoint response = client.post("/analyze/{text}", json=payload_dict) # Assert that the response status code is 200 OK assert response.status_code == 200 # Assert that the sentiment returned is positive assert response.json()[0]['label'] == "positive" # Test the sentiment analysis endpoint for negative sentiment def test_negative_sentiment(): with client: # Define the request payload # Initialize payload as a TextInput object payload = TextInput(text="I'm really disappointed with this service. It's terrible.") # Convert TextInput object to JSON-serializable dictionary payload_dict = jsonable_encoder(payload) # Send a POST request to the sentiment analysis endpoint response = client.post("/analyze/{text}", json=payload_dict) # Assert that the response status code is 200 OK assert response.status_code == 200 # Assert that the sentiment returned is positive assert response.json()[0]['label'] == "negative" # Test the sentiment analysis endpoint for neutral sentiment def test_neutral_sentiment(): with client: # Define the request payload # Initialize payload as a TextInput object payload = TextInput(text="This is a neutral statement.") # Convert TextInput object to JSON-serializable dictionary payload_dict = jsonable_encoder(payload) # Send a POST request to the sentiment analysis endpoint response = client.post("/analyze/{text}", json=payload_dict) # Assert that the response status code is 200 OK assert response.status_code == 200 # Assert that the sentiment returned is positive assert response.json()[0]['label'] == "neutral"