Merge pull request #1 from vaaaaanquish/add_ut

add unittest
vaaaaanquish · Apr 29, 2020 · 1414558 · 1414558
2 parents ce5d848 + d18ff46
commit 1414558
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/unit_test/test_word_data.py b/test/unit_test/test_word_data.py
@@ -6,6 +6,11 @@ class TestCloudia(unittest.TestCase):
     def setUp(self):
         self.cls = WordData('test', [], [], None, None, None, lambda x: [x])
 
+    def test_init_data(self):
+        words, name = self.cls._init_data('test')
+        self.assertListEqual(words, ['test'])
+        self.assertListEqual(name, ['word cloud'])
+
     def test_count(self):
         self.cls.word_num = 2
         self.cls.stop_words = 'test'