diff options
Diffstat (limited to 'boot/test.py')
| -rw-r--r-- | boot/test.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/boot/test.py b/boot/test.py index 1c03a33..c6bd3ff 100644 --- a/boot/test.py +++ b/boot/test.py @@ -12,6 +12,7 @@ sys.path.append(mod_path) import g2p + class TestVocabCompiler(unittest.TestCase): def testWordExtraction(self): @@ -19,7 +20,12 @@ class TestVocabCompiler(unittest.TestCase): dictionary = "temp_dictionary.dic" languagemodel = "temp_languagemodel.lm" - words = ['HACKER', 'LIFE', 'FACEBOOK', 'THIRD', 'NO', 'JOKE', 'NOTIFICATION', 'MEANING', 'TIME', 'TODAY', 'SECOND', 'BIRTHDAY', 'KNOCK KNOCK', 'INBOX', 'OF', 'NEWS', 'YES', 'TOMORROW', 'EMAIL', 'WEATHER', 'FIRST', 'MUSIC', 'SPOTIFY'] + words = [ + 'HACKER', 'LIFE', 'FACEBOOK', 'THIRD', 'NO', 'JOKE', + 'NOTIFICATION', 'MEANING', 'TIME', 'TODAY', 'SECOND', + 'BIRTHDAY', 'KNOCK KNOCK', 'INBOX', 'OF', 'NEWS', 'YES', + 'TOMORROW', 'EMAIL', 'WEATHER', 'FIRST', 'MUSIC', 'SPOTIFY' + ] with patch.object(g2p, 'translateWords') as translateWords: with patch.object(vocabcompiler, 'text2lm') as text2lm: |