LorenzoBioinfo commited on
Commit
665dbc3
·
1 Parent(s): e19093d

Update testapp

Browse files
Files changed (1) hide show
  1. tests/integration/test_app.py +4 -6
tests/integration/test_app.py CHANGED
@@ -22,17 +22,15 @@ def test_predict_endpoint_post():
22
  assert response.status_code == 200
23
  assert any(label in response.text for label in ["positive", "neutral", "negative"])
24
 
25
- @pytest.mark.asyncio
26
  def test_random_tweet_page():
27
  response = client.get("/random_tweet")
28
  assert response.status_code == 200
29
- assert "Sentiment" in response.text
30
-
31
- @pytest.mark.asyncio
32
  def test_random_youtube_page():
33
  response = client.get("/random_youtube_comment")
34
  assert response.status_code == 200
35
- assert "Sentiment" in response.text
36
-
37
 
38
 
 
22
  assert response.status_code == 200
23
  assert any(label in response.text for label in ["positive", "neutral", "negative"])
24
 
 
25
  def test_random_tweet_page():
26
  response = client.get("/random_tweet")
27
  assert response.status_code == 200
28
+ assert any(lbl in response.text for lbl in ["positive", "neutral", "negative", "Positivo", "Neutro", "Negativo"])
29
+
30
+
31
  def test_random_youtube_page():
32
  response = client.get("/random_youtube_comment")
33
  assert response.status_code == 200
34
+ assert any(lbl in response.text for lbl in ["positive", "neutral", "negative", "Positivo", "Neutro", "Negativo"])
 
35
 
36