Add BN tests

This commit is contained in:
Eren Gölge 2023-04-17 13:37:10 +02:00
parent 2d49c05259
commit 2533a18d62
3 changed files with 25 additions and 2 deletions

View File

@ -1,9 +1,9 @@
import re import re
from typing import List from typing import List
import bangla
from bnnumerizer import numerize from bnnumerizer import numerize
from bnunicodenormalizer import Normalizer from bnunicodenormalizer import Normalizer
import bangla
# initialize # initialize
bnorm = Normalizer() bnorm = Normalizer()

View File

@ -52,7 +52,7 @@ class BN_Phonemizer(BasePhonemizer):
if __name__ == "__main__": if __name__ == "__main__":
text = "রাসূলুল্লাহ সাল্লাল্লাহু আলাইহি ওয়া সাল্লাম শিক্ষা দিয়েছেন যে, কেউ যদি কোন খারাপ কিছুর সম্মুখীন হয়, তখনও যেন বলে,।" text = "রাসূলুল্লাহ সাল্লাল্লাহু আলাইহি ওয়া সাল্লাম শিক্ষা দিয়েছেন যে, কেউ যদি কোন খারাপ কিছুর সম্মুখীন হয়, তখনও যেন বলে."
e = BN_Phonemizer() e = BN_Phonemizer()
print(e.supported_languages()) print(e.supported_languages())
print(e.version()) print(e.version())

View File

@ -3,6 +3,7 @@ import unittest
from packaging.version import Version from packaging.version import Version
from TTS.tts.utils.text.phonemizers import ESpeak, Gruut, JA_JP_Phonemizer, ZH_CN_Phonemizer from TTS.tts.utils.text.phonemizers import ESpeak, Gruut, JA_JP_Phonemizer, ZH_CN_Phonemizer
from TTS.tts.utils.text.phonemizers.bangla_phonemizer import BN_Phonemizer
from TTS.tts.utils.text.phonemizers.multi_phonemizer import MultiPhonemizer from TTS.tts.utils.text.phonemizers.multi_phonemizer import MultiPhonemizer
EXAMPLE_TEXTs = [ EXAMPLE_TEXTs = [
@ -230,6 +231,28 @@ class TestZH_CN_Phonemizer(unittest.TestCase):
self.assertTrue(self.phonemizer.is_available()) self.assertTrue(self.phonemizer.is_available())
class TestBN_Phonemizer(unittest.TestCase):
def setUp(self):
self.phonemizer = BN_Phonemizer()
self._TEST_CASES = "রাসূলুল্লাহ সাল্লাল্লাহু আলাইহি ওয়া সাল্লাম শিক্ষা দিয়েছেন যে, কেউ যদি কোন খারাপ কিছুর সম্মুখীন হয়, তখনও যেন"
self._EXPECTED = "রাসূলুল্লাহ সাল্লাল্লাহু আলাইহি ওয়া সাল্লাম শিক্ষা দিয়েছেন যে কেউ যদি কোন খারাপ কিছুর সম্মুখীন হয় তখনও যেন।"
def test_phonemize(self):
self.assertEqual(self.phonemizer.phonemize(self._TEST_CASES, separator=""), self._EXPECTED)
def test_name(self):
self.assertEqual(self.phonemizer.name(), "bn_phonemizer")
def test_get_supported_languages(self):
self.assertIsInstance(self.phonemizer.supported_languages(), dict)
def test_get_version(self):
self.assertIsInstance(self.phonemizer.version(), str)
def test_is_available(self):
self.assertTrue(self.phonemizer.is_available())
class TestMultiPhonemizer(unittest.TestCase): class TestMultiPhonemizer(unittest.TestCase):
def setUp(self): def setUp(self):
self.phonemizer = MultiPhonemizer({"tr": "espeak", "en-us": "", "de": "gruut", "zh-cn": ""}) self.phonemizer = MultiPhonemizer({"tr": "espeak", "en-us": "", "de": "gruut", "zh-cn": ""})