diff options
| -rw-r--r-- | client/test.py | 32 |
1 files changed, 31 insertions, 1 deletions
diff --git a/client/test.py b/client/test.py index 9f23e00..14c13e1 100644 --- a/client/test.py +++ b/client/test.py @@ -11,6 +11,7 @@ import argparse from mock import patch from urllib2 import URLError, urlopen import test_mic +import vocabcompiler import g2p import brain from diagnose import Diagnostics @@ -22,10 +23,38 @@ DEFAULT_PROFILE = { 'phone_number': '012344321' } - def activeInternet(): return Diagnostics.check_network_connection() +class UnorderedList(list): + + def __eq__(self, other): + return sorted(self) == sorted(other) + +class TestVocabCompiler(unittest.TestCase): + + def testWordExtraction(self): + sentences = "temp_sentences.txt" + 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' + ] + + with patch.object(g2p, 'translateWords') as translateWords: + with patch.object(vocabcompiler, 'text2lm') as text2lm: + 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) + os.remove(sentences) + os.remove(dictionary) class TestMic(unittest.TestCase): @@ -208,6 +237,7 @@ if __name__ == '__main__': test_cases = [TestBrain, TestModules] if not args.light: + test_cases.append(TestVocabCompiler) test_cases.append(TestG2P) test_cases.append(TestMic) |