Spaces:

Noveramaaz
/

Sentiment_analysis_docker_API

Sleeping

App Files Files Community

Sentiment_analysis_docker_API / test_main.py

Noveramaaz

Upload 7 files

33761d2 verified 2 months ago

raw history blame contribute delete

No virus

2.66 kB

	from fastapi.testclient import TestClient
	from main import app
	from main 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"