Skip to content

Commit f18b318

Browse files
authored
Add test for sentiment prediction from CSV data
1 parent 5d79f55 commit f18b318

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
import csv
2+
from nlp_model.sentiment_model import predict_sentiment
3+
4+
5+
def load_test_data():
6+
with open("test_data/sentiment_test_data.csv", "r") as file:
7+
reader = csv.DictReader(file)
8+
return list(reader)
9+
10+
11+
def test_sentiment_from_csv():
12+
data = load_test_data()
13+
14+
for row in data:
15+
result = predict_sentiment(row["text"])
16+
assert result == row["expected_sentiment"]

0 commit comments

Comments
 (0)