Update test_main.py

Sibbear1980 · Mar 12, 2024 · 60f9cc1 · 60f9cc1
1 parent c977621
commit 60f9cc1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test_main.py b/test_main.py
@@ -12,7 +12,7 @@ def test_read_main():
 
 def test_predict_positive():
     response = client.post("/predict/",
-                           json={"text": "I like machine learning!"})
+                           json={"text": "I love you!"})
     json_data = response.json()
     assert response.status_code == 200
     assert json_data['Результат:'] == 'позитивный :)'
@@ -23,20 +23,20 @@ def test_predict_negative():
                            json={"text": "I hate machine learning!"})
     json_data = response.json()
     assert response.status_code == 200
-    assert json_data['label'] == 'NEGATIVE'
+    assert json_data['Результат:'] == 'негативный (('
 
 
 def test_predict_russian_positive():
     response = client.post("/predict/",
                            json={"text": "Я люблю учиться"})
     json_data = response.json()
     assert response.status_code == 200
-    assert json_data['label'] == 'POSITIVE'
+    assert json_data['Результат:'] == 'позитивный :)'
 
 
 def test_predict_russian_negative():
     response = client.post("/predict/",
                            json={"text": "Я ненавижу тебя"})
     json_data = response.json()
     assert response.status_code == 200
-    assert json_data['label'] == 'NEGATIVE'
+    assert json_data['Результат:'] == 'негативный (('