mirror of https://github.com/coqui-ai/TTS.git
Add BN tests
This commit is contained in:
parent
2d49c05259
commit
2533a18d62
|
@ -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()
|
||||||
|
|
|
@ -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())
|
||||||
|
|
|
@ -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": ""})
|
||||||
|
|
Loading…
Reference in New Issue