Update test_main.py

Sibbear1980 · Mar 12, 2024 · 061f2eb · 061f2eb
1 parent 788ce6b
commit 061f2eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_main.py b/test_main.py
@@ -15,7 +15,7 @@ def test_predict_positive():
                            json={"text": "I like machine learning!"})
     json_data = response.json()
     assert response.status_code == 200
-    assert json_data['label'] == 'Текст позитивный'
+    assert json_data['text'] == 'Текст позитивный'
 
 
 def test_predict_negative():