43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import unittest
|
|
|
|
from argparse import Namespace
|
|
from deepspeech_training.util.importers import validate_label_eng, get_validate_label
|
|
from pathlib import Path
|
|
|
|
def from_here(path):
|
|
here = Path(__file__)
|
|
return here.parent / path
|
|
|
|
class TestValidateLabelEng(unittest.TestCase):
|
|
def test_numbers(self):
|
|
label = validate_label_eng("this is a 1 2 3 test")
|
|
self.assertEqual(label, None)
|
|
|
|
class TestGetValidateLabel(unittest.TestCase):
|
|
|
|
def test_no_validate_label_locale(self):
|
|
f = get_validate_label(Namespace())
|
|
self.assertEqual(f('toto'), 'toto')
|
|
self.assertEqual(f('toto1234'), None)
|
|
self.assertEqual(f('toto1234[{[{[]'), None)
|
|
|
|
def test_validate_label_locale_default(self):
|
|
f = get_validate_label(Namespace(validate_label_locale=None))
|
|
self.assertEqual(f('toto'), 'toto')
|
|
self.assertEqual(f('toto1234'), None)
|
|
self.assertEqual(f('toto1234[{[{[]'), None)
|
|
|
|
def test_get_validate_label_missing(self):
|
|
args = Namespace(validate_label_locale=from_here('test_data/validate_locale_ger.py'))
|
|
f = get_validate_label(args)
|
|
self.assertEqual(f, None)
|
|
|
|
def test_get_validate_label(self):
|
|
args = Namespace(validate_label_locale=from_here('test_data/validate_locale_fra.py'))
|
|
f = get_validate_label(args)
|
|
l = f('toto')
|
|
self.assertEqual(l, 'toto')
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|