update tests

meedan · Aug 29, 2023 · 2ea48ee · 2ea48ee
1 parent 775e7fc
commit 2ea48ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/lib/model/test_fasttext.py b/test/lib/model/test_fasttext.py
@@ -18,8 +18,8 @@ def test_respond(self):
         response = self.model.respond(query)
 
         self.assertEqual(len(response), 2)
-        self.assertEqual(response[0].response, "en")
-        self.assertEqual(response[1].response, "zh-Hans")
+        self.assertEqual(response[0].response, {'language': 'en', 'script': None, 'score': 1.0000062})
+        self.assertEqual(response[1].response, {'language': 'zh', 'script': 'Hans', 'score': 0.83046132})
 
 if __name__ == '__main__':
     unittest.main()