feat: Updated tests/test_jieba.py

Sygil-Dev · Jan 3, 2024 · a158df6 · a158df6
1 parent 77d7d1b
commit a158df6
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_jieba.py b/tests/test_jieba.py
@@ -8,16 +8,16 @@ def setUp(self):
         self.text = "This is a test text for jieba"
 
     def test_cut(self):
-        result = jieba.cut(self.text)
-        self.assertEqual(list(result), ['This', ' ', 'is', ' ', 'a', ' ', 'test', ' ', 'text', ' ', 'for', ' ', 'jieba'])
+        self._test_cut_method(jieba.cut, ['This', ' ', 'is', ' ', 'a', ' ', 'test', ' ', 'text', ' ', 'for', ' ', 'jieba'])
 
     def test_cut_all(self):
-        result = jieba.cut(self.text, cut_all=True)
-        self.assertEqual(list(result), ['This', ' ', 'is', ' ', 'a', ' ', 'test', ' ', 'text', ' ', 'for', ' ', 'jieba'])
+        self._test_cut_method(jieba.cut, ['This', ' ', 'is', ' ', 'a', ' ', 'test', ' ', 'text', ' ', 'for', ' ', 'jieba'], cut_all=True)
 
     def test_cut_for_search(self):
-        result = jieba.cut_for_search(self.text)
-        self.assertEqual(list(result), ['This', ' ', 'is', ' ', 'a', ' ', 'test', ' ', 'text', ' ', 'for', ' ', 'jieba'])
+        self._test_cut_method(jieba.cut_for_search, ['This', ' ', 'is', ' ', 'a', ' ', 'test', ' ', 'text', ' ', 'for', ' ', 'jieba'])
 
 if __name__ == '__main__':
     unittest.main()
+    def _test_cut_method(self, method, expected_result, **kwargs):
+        result = method(self.text, **kwargs)
+        self.assertEqual(list(result), expected_result)