Add skip conditions to unittests for dependency checking

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/test_stt.py b/tests/test_stt.py
index 267c1c2..cb33ff6 100644
--- a/tests/test_stt.py
+++ b/tests/test_stt.py
@@ -1,9 +1,30 @@
 #!/usr/bin/env python2
 # -*- coding: utf-8-*-
 import unittest
+import imp
 from client import stt, jasperpath
 
 
+def cmuclmtk_installed():
+    try:
+        imp.find_module('cmuclmtk')
+    except ImportError:
+        return False
+    else:
+        return True
+
+
+def pocketsphinx_installed():
+    try:
+        imp.find_module('pocketsphinx')
+    except ImportError:
+        return False
+    else:
+        return True
+
+
+@unittest.skipUnless(cmuclmtk_installed(), "CMUCLMTK not present")
+@unittest.skipUnless(pocketsphinx_installed(), "Pocketsphinx not present")
 class TestSTT(unittest.TestCase):
 
     def setUp(self):