test: Add main_test.py (#23)

.github/workflows/ci.yml

@@ -19,3 +19,4 @@ jobs:
       - uses: actions/checkout@v2.3.1
       - run: docker-compose build
       - run: make build
+      - run: make test

.gitignore

@@ -1,2 +1,3 @@
 .DS_Store
 build
+__pycache__

Makefile

@@ -1,3 +1,12 @@
 .PHONY: build
 build:
 	docker-compose run --rm -T dev python main.py
+
+.PHONY: test
+test:
+	docker-compose run --rm -T dev python main_test.py
+
+.PHONY: all
+all:
+	$(MAKE) test
+	$(MAKE) build

main.py

@@ -7,6 +7,22 @@
 EMOJI_JSON_URL = 'https://raw.githubusercontent.com/yagays/emoji-ja/20190726/data/emoji_ja.json'
 EMOJI_DICT_PATH = 'tsv/emoji.tsv'
 kakasi = kakasi()
+kakasi.setMode("J","H")
+conv_j2h = kakasi.getConverter()
+kakasi.setMode("K","H")
+conv_k2h = kakasi.getConverter()
+
+
+def hiraganafy(keyword):
+    k = keyword.upper()
+    k = conv_j2h.do(k)
+    k = conv_k2h.do(k)
+    return k
+
+
+def add_word_to_dict(emoji, keyword, emoji_dict):
+    word = f':{keyword}\t{emoji}\t記号\t'
+    emoji_dict.append(word)
 
 
 class EmojiDict():
@@ -31,28 +47,14 @@ def create_emoji_dict(self) -> None:
             for k in self.emoji_json[emoji]['keywords']:
                 if k.isalpha() is False:
                     continue
-                k = k.upper()
-                kakasi.setMode("J","H")
-                conv = kakasi.getConverter()
-                k = conv.do(k)
-                kakasi.setMode("K","H")
-                conv = kakasi.getConverter()
-                k = conv.do(k)
-                w = f':{k}\t{emoji}\t記号\t'
-                self.emoji_dict.append(w)
+                k = hiraganafy(k)
+                add_word_to_dict(emoji, k, self.emoji_dict)
 
             k = self.emoji_json[emoji]['short_name']
             if k.isalpha() is False:
                 continue
-            k = k.upper()
-            kakasi.setMode("J","H")
-            conv = kakasi.getConverter()
-            k = conv.do(k)
-            kakasi.setMode("K","H")
-            conv = kakasi.getConverter()
-            k = conv.do(k)
-            w = f':{k}\t{emoji}\t記号\t'
-            self.emoji_dict.append(w)
+            k = hiraganafy(k)
+            add_word_to_dict(emoji, k, self.emoji_dict)
 
         self.emoji_dict.sort()
         self.emoji_dict = '\n'.join(self.emoji_dict)

main_test.py

@@ -0,0 +1,22 @@
+import unittest
+import main
+
+class TestStringMethods(unittest.TestCase):
+
+    def test_hiraganafy(self):
+        katakana = 'カタカナ'
+        self.assertEqual(main.hiraganafy(katakana), 'かたかな')
+        kanji = '漢字'
+        self.assertEqual(main.hiraganafy(kanji), 'かんじ')
+        hiragana = 'ひらがな'
+        self.assertEqual(main.hiraganafy(hiragana), 'ひらがな')
+
+    def test_add_word_to_dict(self):
+        emoji_dict = []
+        main.add_word_to_dict('🇯🇵', 'にっぽん', emoji_dict)
+        expected_row = f':にっぽん\t🇯🇵\t記号\t'
+        self.assertEqual(expected_row, emoji_dict[0])
+
+
+if __name__ == '__main__':
+    unittest.main(verbosity=2)