Update summarize test

Author: bact

README.md

@@ -80,7 +80,7 @@ pip install pythainlp[extra1,extra2,...]
 
 where `extras` can be
   - `attacut` (to support attacut, a fast and accurate tokenizer)
-  - `benchmarks` (for word tokenization benchmarking)
+  - `benchmarks` (for [word tokenization benchmarking](tokenization-benchmark.md))
   - `icu` (for ICU, International Components for Unicode, support in transliteration and tokenization)
   - `ipa` (for IPA, International Phonetic Alphabet, support in transliteration)
   - `ml` (to support ULMFiT models for classification)
@@ -180,7 +180,7 @@ pip install pythainlp[extra1,extra2,...]
 
 โดยที่ `extras` คือ
   - `attacut` (ตัวตัดคำที่แม่นกว่า `newmm` เมื่อเทียบกับชุดข้อมูล BEST)
-  - `benchmarks` (สำหรับเครื่องมือวัดความแม่นยำของตัวตัดคำ)
+  - `benchmarks` (สำหรับเครื่องมือ[วัดความแม่นยำของตัวตัดคำ](tokenization-benchmark.md))
   - `icu` (สำหรับการถอดตัวสะกดเป็นสัทอักษรและการตัดคำด้วย ICU)
   - `ipa` (สำหรับการถอดตัวสะกดเป็นสัทอักษรสากล (IPA))
   - `ml` (สำหรับการรองรับโมเดล ULMFiT)

pythainlp/summarize/__init__.py

@@ -7,6 +7,6 @@
     "summarize",
 ]
 
-DEFAULT_SUMMARIZE_ENGINE = "freq"
+DEFAULT_SUMMARIZE_ENGINE = "frequency"
 
 from pythainlp.summarize.core import summarize

tests/test_summarize.py

@@ -6,17 +6,18 @@
 
 
 class TestSummarizePackage(unittest.TestCase):
-
     def test_summarize(self):
-        text = "อาหาร หมายถึง ของแข็งหรือของเหลว "
-        text += "ที่กินหรือดื่มเข้าสู่ร่างกายแล้ว "
-        text += "จะทำให้เกิดพลังงานและความร้อนแก่ร่างกาย "
-        text += "ทำให้ร่างกายเจริญเติบโต "
-        text += "ซ่อมแซมส่วนที่สึกหรอ ควบคุมการเปลี่ยนแปลงต่างๆ ในร่างกาย "
-        text += "ช่วยทำให้อวัยวะต่างๆ ทำงานได้อย่างปกติ "
-        text += "อาหารจะต้องไม่มีพิษและไม่เกิดโทษต่อร่างกาย"
+        text = (
+            "อาหาร หมายถึง ของแข็งหรือของเหลว "
+            "ที่กินหรือดื่มเข้าสู่ร่างกายแล้ว "
+            "จะทำให้เกิดพลังงานและความร้อนแก่ร่างกาย "
+            "ทำให้ร่างกายเจริญเติบโต "
+            "ซ่อมแซมส่วนที่สึกหรอ ควบคุมการเปลี่ยนแปลงต่างๆ ในร่างกาย "
+            "ช่วยทำให้อวัยวะต่างๆ ทำงานได้อย่างปกติ "
+            "อาหารจะต้องไม่มีพิษและไม่เกิดโทษต่อร่างกาย"
+        )
         self.assertEqual(
-            summarize(text=text, n=1, engine="frequency"),
+            summarize(text=text, n=1),
             ["อาหารจะต้องไม่มีพิษและไม่เกิดโทษต่อร่างกาย"],
         )
         self.assertIsNotNone(summarize(text, 1, engine="XX"))