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
from typing import List
import bangla
from bnnumerizer import numerize
from bnunicodenormalizer import Normalizer
import bangla
# initialize
bnorm = Normalizer()

View File

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

View File

@ -3,6 +3,7 @@ import unittest
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.bangla_phonemizer import BN_Phonemizer
from TTS.tts.utils.text.phonemizers.multi_phonemizer import MultiPhonemizer
EXAMPLE_TEXTs = [
@ -230,6 +231,28 @@ class TestZH_CN_Phonemizer(unittest.TestCase):
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):
def setUp(self):
self.phonemizer = MultiPhonemizer({"tr": "espeak", "en-us": "", "de": "gruut", "zh-cn": ""})