Spaces:
Sleeping
Sleeping
docstring
Browse files- tests/model_tests.py +1 -0
tests/model_tests.py
CHANGED
@@ -10,6 +10,7 @@ class TestClassifier(unittest.TestCase):
|
|
10 |
'Libya', 'Bahrain', 'Qatar', 'Yemen', 'Palestine', 'Jordan', 'Somalia', 'Sudan']
|
11 |
|
12 |
def test_output(self):
|
|
|
13 |
request_data = {"text": "حاجة حلوة اكيد"}
|
14 |
response = requests.post(self.API_URL, json=request_data)
|
15 |
self.assertEqual(response.status_code, 200)
|
|
|
10 |
'Libya', 'Bahrain', 'Qatar', 'Yemen', 'Palestine', 'Jordan', 'Somalia', 'Sudan']
|
11 |
|
12 |
def test_output(self):
|
13 |
+
"""Test if the response of the /classify API endpoint is correct"""
|
14 |
request_data = {"text": "حاجة حلوة اكيد"}
|
15 |
response = requests.post(self.API_URL, json=request_data)
|
16 |
self.assertEqual(response.status_code, 200)
|