add nagisa.utils test to nagisa_test.py

taishi-i · Jul 4, 2020 · 21961ea · 21961ea
1 parent ab4d4a7
commit 21961ea
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/nagisa_test.py b/test/nagisa_test.py
@@ -126,9 +126,12 @@ def test_utils(self):
         self.assertEqual(output, nagisa.utils.PAD)
 
         # test_22
-        text = "Python"
-        output = ['P', 'y', 't', 'h', 'o', 'n']
+        text = "Ｐｙｔｈｏｎ"
+        text = nagisa.utils.preprocess(text)
+        text = text.lower()
+        output = ['p', 'y', 't', 'h', 'o', 'n']
         unigrams = nagisa.utils.get_unigram(text)
+
         self.assertEqual(output, unigrams)
 
     def test_fit(self):