diff options
| -rw-r--r-- | client/test.py | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/client/test.py b/client/test.py index 7366ef2..6eaaa5a 100644 --- a/client/test.py +++ b/client/test.py @@ -13,6 +13,7 @@ import g2p import brain import jasperpath import tts +from stt import TranscriptionMode from diagnose import Diagnostics DEFAULT_PROFILE = { @@ -68,14 +69,20 @@ class TestMic(unittest.TestCase): self.stt = PocketSphinxSTT() def testTranscribeJasper(self): - """Does Jasper recognize his name (i.e., passive listen)?""" - transcription = self.stt.transcribe(self.jasper_clip, - PERSONA_ONLY=True) + """ + Does Jasper recognize his name (i.e., passive listen)? + """ + with open(self.jasper_clip, mode="rb") as f: + transcription = self.stt.transcribe(f, + mode=TranscriptionMode.KEYWORD) self.assertTrue("JASPER" in transcription) def testTranscribe(self): - """Does Jasper recognize 'time' (i.e., active listen)?""" - transcription = self.stt.transcribe(self.time_clip) + """ + Does Jasper recognize 'time' (i.e., active listen)? + """ + with open(self.time_clip, mode="rb") as f: + transcription = self.stt.transcribe(f) self.assertTrue("TIME" in transcription) |