zaidmehdi commited on
Commit
464be69
1 Parent(s): 9480e34
Files changed (1) hide show
  1. 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)