more test cases for thai_time()

Author: bact

tests/test_tokenize.py

@@ -235,7 +235,7 @@ def test_subword_tokenize(self):
 
         self.assertEqual(subword_tokenize(None, engine="ssg"), [])
         self.assertEqual(subword_tokenize("", engine="ssg"), [])
-        self.assertIsNotNone(subword_tokenize("สวัสดีดาวอังคาร", engine="ssg"))
+        self.assertTrue("ดาว" in subword_tokenize("สวัสดีดาวอังคาร", engine="ssg"))
 
     def test_syllable_tokenize(self):
         self.assertEqual(syllable_tokenize(None), [])

tests/test_util.py

@@ -168,10 +168,10 @@ def test_thai_strftime(self):
     # ### pythainlp.util.thai_time
 
     def test_thai_time(self):
+        self.assertEqual(thai_time("8:17"), thai_time("08:17"))
         self.assertEqual(thai_time("8:17"), "แปดนาฬิกาสิบเจ็ดนาที")
         self.assertEqual(thai_time("8:17", "6h"), "สองโมงเช้าสิบเจ็ดนาที")
         self.assertEqual(thai_time("8:17", "m6h"), "แปดโมงสิบเจ็ดนาที")
-        self.assertEqual(thai_time("18:30", "m6h"), "หกโมงครึ่ง")
         self.assertEqual(thai_time("13:30:01", "6h", "m"), "บ่ายโมงครึ่ง")
         self.assertEqual(
             thai_time(datetime.time(12, 3, 0)), "สิบสองนาฬิกาสามนาที"
@@ -181,23 +181,38 @@ def test_thai_time(self):
             "สิบสองนาฬิกาสามนาทีหนึ่งวินาที",
         )
         self.assertEqual(
-            thai_time(
-                datetime.datetime(2014, 5, 22, 12, 3, 0), precision="s"
-            ),
+            thai_time(datetime.datetime(2014, 5, 22, 12, 3, 0), precision="s"),
             "สิบสองนาฬิกาสามนาทีศูนย์วินาที",
         )
         self.assertEqual(
-            thai_time(
-                datetime.datetime(2014, 5, 22, 12, 3, 1), precision="m"
-            ),
+            thai_time(datetime.datetime(2014, 5, 22, 12, 3, 1), precision="m"),
             "สิบสองนาฬิกาสามนาที",
         )
         self.assertEqual(
-            thai_time(
-                datetime.datetime(1976, 10, 6, 12, 30, 1), "6h", "m"
-            ),
+            thai_time(datetime.datetime(1976, 10, 6, 12, 30, 1), "6h", "m"),
             "เที่ยงครึ่ง",
         )
+        self.assertEqual(thai_time("18:30"), "สิบแปดนาฬิกาสามสิบนาที")
+        self.assertEqual(thai_time("18:30:00"), "สิบแปดนาฬิกาสามสิบนาที")
+        self.assertEqual(
+            thai_time("18:30:01"), "สิบแปดนาฬิกาสามสิบนาทีหนึ่งวินาที"
+        )
+        self.assertEqual(
+            thai_time("18:30:01", precision="m"), "สิบแปดนาฬิกาสามสิบนาที"
+        )
+        self.assertEqual(
+            thai_time("18:30:01", precision="s"),
+            "สิบแปดนาฬิกาสามสิบนาทีหนึ่งวินาที",
+        )
+        self.assertEqual(
+            thai_time("18:30:01", fmt="m6h", precision="m"), "หกโมงครึ่ง"
+        )
+        self.assertEqual(
+            thai_time("18:30:01", fmt="m6h"), "หกโมงสามสิบนาทีหนึ่งวินาที"
+        )
+        self.assertEqual(
+            thai_time("18:30:01", fmt="m6h", precision="m"), "หกโมงครึ่ง"
+        )
         self.assertIsNotNone(thai_time("0:30"))
         self.assertIsNotNone(thai_time("0:30", "6h"))
         self.assertIsNotNone(thai_time("0:30", "m6h"))