import unittest
import bot.entries.base.expressions as Expressions


class TestExpressions(unittest.TestCase):
    def test_kata_to_hira(self):
        hira = "Abc5゠ぁゖずほヷヸヹヺ・ーゝゞヿ"
        kata = "Abc5゠ァヶズホヷヸヹヺ・ーヽヾヿ"
        transformed = Expressions.kata_to_hira(kata)
        self.assertEqual(transformed, hira)

    def test_add_fullwidth(self):
        exps = ["Abc059!~{}あア日本語Ａｂｃ０５９！～｛｝"]
        Expressions.add_fullwidth(exps)
        self.assertEqual(len(exps), 2)
        self.assertIn("Abc059!~{}あア日本語Ａｂｃ０５９！～｛｝", exps)
        self.assertIn("Ａｂｃ０５９！～｛｝あア日本語Ａｂｃ０５９！～｛｝", exps)

    def test_add_iteration_mark(self):
        exps = ["禍禍しい", "凶々しい", "凶凶しい"]
        Expressions.add_iteration_mark(exps)
        self.assertEqual(len(exps), 4)
        self.assertIn("禍々しい", exps)
        self.assertIn("禍禍しい", exps)
        self.assertIn("凶々しい", exps)
        self.assertIn("凶凶しい", exps)

    def test_remove_iteration_mark(self):
        exps = ["禍々しい", "凶々しい", "凶凶しい"]
        Expressions.remove_iteration_mark(exps)
        self.assertEqual(len(exps), 4)
        self.assertIn("禍々しい", exps)
        self.assertIn("禍禍しい", exps)
        self.assertIn("凶々しい", exps)
        self.assertIn("凶凶しい", exps)

    def test_add_variant_kanji1(self):
        exps = ["剥く", "摑む"]
        Expressions.add_variant_kanji(exps)
        self.assertEqual(len(exps), 4)
        self.assertIn("剥く", exps)
        self.assertIn("剝く", exps)
        self.assertIn("掴む", exps)
        self.assertIn("摑む", exps)

    def test_add_variant_kanji2(self):
        exps = ["剝く", "掴む", "摑む"]
        Expressions.add_variant_kanji(exps)
        self.assertEqual(len(exps), 4)
        self.assertIn("剥く", exps)
        self.assertIn("剝く", exps)
        self.assertIn("掴む", exps)
        self.assertIn("摑む", exps)

    def test_add_variant_kanji3(self):
        exps = ["剝摑"]
        Expressions.add_variant_kanji(exps)
        self.assertEqual(len(exps), 4)
        self.assertIn("剝摑", exps)
        self.assertIn("剝掴", exps)
        self.assertIn("剥掴", exps)
        self.assertIn("剥摑", exps)

    def test_add_variant_kanji4(self):
        exps = ["剥掴"]
        Expressions.add_variant_kanji(exps)
        self.assertEqual(len(exps), 4)
        self.assertIn("剝摑", exps)
        self.assertIn("剝掴", exps)
        self.assertIn("剥掴", exps)
        self.assertIn("剥摑", exps)

    def test_expand_abbreviation(self):
        text = "有（り）合（わ）せ"
        abbrs = Expressions.expand_abbreviation(text)
        self.assertEqual(len(abbrs), 4)
        self.assertIn("有り合わせ", abbrs)
        self.assertIn("有合わせ", abbrs)
        self.assertIn("有り合せ", abbrs)
        self.assertIn("有合せ", abbrs)

    def test_expand_abbreviation_list(self):
        texts = ["有（り）合わせ", "有り合（わ）せ", "有合せ"]
        abbrs = Expressions.expand_abbreviation_list(texts)
        self.assertEqual(len(abbrs), 4)
        self.assertIn("有り合わせ", abbrs)
        self.assertIn("有合わせ", abbrs)
        self.assertIn("有り合せ", abbrs)
        self.assertIn("有合せ", abbrs)