Add test for sentiment analysis

CHANGELOG.md

@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - Added regression tests for training configs that run on a scheduled workflow.
+- Added a test for the pretrained sentiment analysis model.
 
 ## [v1.1.0rc3](https:/allenai/allennlp-models/releases/tag/v1.1.0rc3) - 2020-08-12
 

tests/pretrained_test.py

@@ -361,6 +361,11 @@ def test_dependency_parsing(self):
  ]
  assert result["predicted_heads"] == [2, 0, 2, 2, 4, 2]
 
+ def test_sentiment_analysis(self):
+ predictor = load_predictor("roberta-sst")
+ result = predictor.predict_json({"sentence": "This is a positive review."})
+ assert result["label"] == "1"
+
  def test_openie(self):
  predictor = load_predictor("structured-prediction-srl")
  result = predictor.predict_json(