test if deprecated function raises DeprecationWarning

Author: bact

tests/test_tokenize.py

@@ -26,9 +26,6 @@
 
 
 class TestTokenizePackage(unittest.TestCase):
-    def test_dict_word_tokenize(self):
-        self.assertEqual(dict_word_tokenize(""), [])
-
     def test_etcc(self):
         self.assertEqual(etcc.segment(""), "")
         self.assertIsInstance(etcc.segment("คืนความสุข"), list)
@@ -81,6 +78,9 @@ def test_word_tokenize(self):
             word_tokenize("รถไฟฟ้าBTS", custom_dict=DEFAULT_DICT_TRIE)
         )
 
+        with self.assertRaises(DeprecationWarning):
+            dict_word_tokenize("เลิกใช้แล้ว")
+
     def test_Tokenizer(self):
         t_test = Tokenizer(DEFAULT_DICT_TRIE)
         self.assertEqual(t_test.word_tokenize(""), [])

tests/test_util.py

@@ -243,7 +243,9 @@ def test_thai_time(self):
     def test_delete_tone(self):
         self.assertEqual(delete_tone("จิ้น"), "จิน")
         self.assertEqual(delete_tone("เก๋า"), "เกา")
-        self.assertEqual(delete_tone("จิ้น"), deletetone("จิ้น"))
+
+        with self.assertRaises(DeprecationWarning):
+            deletetone("จิ้น")
 
     def test_normalize(self):
         self.assertEqual(normalize("เเปลก"), "แปลก")
@@ -271,7 +273,6 @@ def test_isthai(self):
         self.assertEqual(isthai("(ต.ค.)", ignore_chars=".()"), True)
 
     def test_is_native_thai(self):
-        self.assertEqual(is_native_thai("เลข"), thaicheck("เลข"))
         self.assertEqual(is_native_thai(None), False)
         self.assertEqual(is_native_thai(""), False)
         self.assertEqual(is_native_thai("116"), False)
@@ -291,3 +292,6 @@ def test_is_native_thai(self):
         self.assertEqual(is_native_thai("เลข"), False)
         self.assertEqual(is_native_thai("เทเวศน์"), False)
         self.assertEqual(is_native_thai("เทเวศร์"), False)
+
+        with self.assertRaises(DeprecationWarning):
+            thaicheck("เลข")