Changed vocabcompiler test cases

Author: Holzhaus

client/test.py

@@ -3,6 +3,8 @@
 import os
 import sys
 import unittest
+import tempfile
+import shutil
 import argparse
 from mock import patch
 from urllib2 import URLError, urlopen
@@ -31,28 +33,30 @@ def __eq__(self, other):
 
 class TestVocabCompiler(unittest.TestCase):
 
-    def testWordExtraction(self):
-        sentences = "temp_sentences.txt"
-        dictionary = "temp_dictionary.dic"
-        languagemodel = "temp_languagemodel.lm"
-
-        words = [
+    def testPhraseExtraction(self):
+        expected_phrases = sorted([
             '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)
+        ])
+        
+        extracted_phrases = vocabcompiler.get_all_phrases()
+        self.assertTrue(extracted_phrases == extracted_phrases)
+    
+    def testVocabulary(self):
+        phrases = ['THIS IS A TEST']
+
+        tempdir = tempfile.mkdtemp()
+
+        vocab = vocabcompiler.DummyVocabulary(path=tempdir)
+        self.assertTrue(vocab.compiled_revision is None)
+        self.assertTrue(not vocab.is_compiled)
+        vocab.compile(phrases)
+        self.assertTrue(vocab.is_compiled)
+        self.assertTrue(vocab.matches_phrases(phrases))
+
+        shutil.rmtree(tempdir)
 
 class TestMic(unittest.TestCase):