Split unittests into separate files and move them to tests/

1 files changed, 30 insertions, 0 deletions

diff --git a/tests/test_stt.py b/tests/test_stt.py
new file mode 100644
index 0000000..267c1c2
--- /dev/null
+++ b/tests/test_stt.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python2
+# -*- coding: utf-8-*-
+import unittest
+from client import stt, jasperpath
+
+
+class TestSTT(unittest.TestCase):
+
+    def setUp(self):
+        self.jasper_clip = jasperpath.data('audio', 'jasper.wav')
+        self.time_clip = jasperpath.data('audio', 'time.wav')
+
+        self.passive_stt_engine = stt.PocketSphinxSTT.get_passive_instance()
+        self.active_stt_engine = stt.PocketSphinxSTT.get_active_instance()
+
+    def testTranscribeJasper(self):
+        """
+        Does Jasper recognize his name (i.e., passive listen)?
+        """
+        with open(self.jasper_clip, mode="rb") as f:
+            transcription = self.passive_stt_engine.transcribe(f)
+        self.assertIn("JASPER", transcription)
+
+    def testTranscribe(self):
+        """
+        Does Jasper recognize 'time' (i.e., active listen)?
+        """
+        with open(self.time_clip, mode="rb") as f:
+            transcription = self.active_stt_engine.transcribe(f)
+        self.assertIn("TIME", transcription)