diff options
Diffstat (limited to 'client')
| -rw-r--r-- | client/test.py | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/client/test.py b/client/test.py index 184a777..0ad37a3 100644 --- a/client/test.py +++ b/client/test.py @@ -4,7 +4,7 @@ import os import sys import unittest import argparse -from mock import patch +from mock import patch, Mock from urllib2 import URLError, urlopen import test_mic @@ -37,20 +37,23 @@ class TestVocabCompiler(unittest.TestCase): 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' + 'MUSIC', 'SPOTIFY' ] + mock_module = Mock() + mock_module.WORDS = [ + 'MOCK' + ] + + words.extend(mock_module.WORDS) + with patch.object(g2p, 'translateWords') as translateWords: with patch.object(vocabcompiler, 'text2lm') as text2lm: - vocabcompiler.compile(sentences, dictionary, languagemodel) + with patch.object(brain.Brain, 'get_modules', classmethod(lambda cls: [mock_module])) as modules: + vocabcompiler.compile(sentences, dictionary, languagemodel) - # 'words' is appended with ['MUSIC', 'SPOTIFY'] - # so must be > 2 to have received WORDS from modules - translateWords.assert_called_once_with(UnorderedList(words)) - self.assertTrue(text2lm.called) + translateWords.assert_called_once_with(UnorderedList(words)) + self.assertTrue(text2lm.called) os.remove(sentences) os.remove(dictionary) |