35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
import speech_recognition as sr
|
|
|
|
|
|
class TestSpecialFeatures(unittest.TestCase):
|
|
def setUp(self):
|
|
self.AUDIO_FILE_EN = os.path.join(os.path.dirname(os.path.realpath(__file__)), "english.wav")
|
|
self.addTypeEqualityFunc(str, self.assertSameWords)
|
|
|
|
@unittest.skipIf(sys.platform.startswith("win"), "skip on Windows")
|
|
def test_sphinx_keywords(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_EN) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_sphinx(audio, keyword_entries=[("one", 1.0), ("two", 1.0), ("three", 1.0)]), "three two one")
|
|
# pocketsphinx < 5 recognizes tree but pocketsphinx >= 5 ignores it (TODO need to research why)
|
|
self.assertEqual(r.recognize_sphinx(audio, keyword_entries=[("wan", 0.95), ("too", 1.0), ("tree", 1.0)]), "too wan")
|
|
# pocketsphinx < 5 recognizes tee but pocketsphinx >= 5 ignores it (TODO need to research why)
|
|
self.assertEqual(r.recognize_sphinx(audio, keyword_entries=[("un", 0.95), ("to", 1.0), ("tee", 1.0)]), "to un")
|
|
|
|
def assertSameWords(self, tested, reference, msg=None):
|
|
set_tested = set(tested.split())
|
|
set_reference = set(reference.split())
|
|
if set_tested != set_reference:
|
|
raise self.failureException(msg if msg is not None else "%r doesn't consist of the same words as %r" % (tested, reference))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|