89 lines
5.0 KiB
Python
89 lines
5.0 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
import speech_recognition as sr
|
|
|
|
|
|
class TestRecognition(unittest.TestCase):
|
|
def setUp(self):
|
|
self.AUDIO_FILE_EN = os.path.join(os.path.dirname(os.path.realpath(__file__)), "english.wav")
|
|
self.AUDIO_FILE_FR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "french.aiff")
|
|
self.AUDIO_FILE_ZH = os.path.join(os.path.dirname(os.path.realpath(__file__)), "chinese.flac")
|
|
|
|
def test_recognizer_attributes(self):
|
|
r = sr.Recognizer()
|
|
attributes = set(dir(r))
|
|
|
|
self.assertEqual(r.energy_threshold, 300)
|
|
self.assertTrue(r.dynamic_energy_threshold)
|
|
self.assertEqual(r.dynamic_energy_adjustment_damping, 0.15)
|
|
self.assertEqual(r.dynamic_energy_ratio, 1.5)
|
|
self.assertEqual(r.pause_threshold, 0.8)
|
|
self.assertIsNone(r.operation_timeout)
|
|
self.assertEqual(r.phrase_threshold, 0.3)
|
|
self.assertEqual(r.non_speaking_duration, 0.5)
|
|
# https://github.com/Uberi/speech_recognition/issues/743
|
|
self.assertTrue("recognize_google" in attributes)
|
|
|
|
@unittest.skipIf(sys.platform.startswith("win"), "skip on Windows")
|
|
def test_sphinx_english(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_EN) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_sphinx(audio), "one two three")
|
|
|
|
@unittest.skipUnless("WIT_AI_KEY" in os.environ, "requires Wit.ai key to be specified in WIT_AI_KEY environment variable")
|
|
def test_wit_english(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_EN) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_wit(audio, key=os.environ["WIT_AI_KEY"]), "one two three")
|
|
|
|
@unittest.skipUnless("BING_KEY" in os.environ, "requires Microsoft Bing Voice Recognition key to be specified in BING_KEY environment variable")
|
|
def test_bing_english(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_EN) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_bing(audio, key=os.environ["BING_KEY"]), "123.")
|
|
|
|
@unittest.skipUnless("BING_KEY" in os.environ, "requires Microsoft Bing Voice Recognition key to be specified in BING_KEY environment variable")
|
|
def test_bing_french(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_FR) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_bing(audio, key=os.environ["BING_KEY"], language="fr-FR"), u"Essaye la dictée numéro un.")
|
|
|
|
@unittest.skipUnless("BING_KEY" in os.environ, "requires Microsoft Bing Voice Recognition key to be specified in BING_KEY environment variable")
|
|
def test_bing_chinese(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_ZH) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_bing(audio, key=os.environ["BING_KEY"], language="zh-CN"), u"砸自己的脚。")
|
|
|
|
@unittest.skipUnless("HOUNDIFY_CLIENT_ID" in os.environ and "HOUNDIFY_CLIENT_KEY" in os.environ, "requires Houndify client ID and client key to be specified in HOUNDIFY_CLIENT_ID and HOUNDIFY_CLIENT_KEY environment variables")
|
|
def test_houndify_english(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_EN) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_houndify(audio, client_id=os.environ["HOUNDIFY_CLIENT_ID"], client_key=os.environ["HOUNDIFY_CLIENT_KEY"]), "one two three")
|
|
|
|
@unittest.skipUnless("IBM_USERNAME" in os.environ and "IBM_PASSWORD" in os.environ, "requires IBM Speech to Text username and password to be specified in IBM_USERNAME and IBM_PASSWORD environment variables")
|
|
def test_ibm_english(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_EN) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_ibm(audio, username=os.environ["IBM_USERNAME"], password=os.environ["IBM_PASSWORD"]), "one two three ")
|
|
|
|
@unittest.skipUnless("IBM_USERNAME" in os.environ and "IBM_PASSWORD" in os.environ, "requires IBM Speech to Text username and password to be specified in IBM_USERNAME and IBM_PASSWORD environment variables")
|
|
def test_ibm_french(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_FR) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_ibm(audio, username=os.environ["IBM_USERNAME"], password=os.environ["IBM_PASSWORD"], language="fr-FR"), u"si la dictée numéro un ")
|
|
|
|
@unittest.skipUnless("IBM_USERNAME" in os.environ and "IBM_PASSWORD" in os.environ, "requires IBM Speech to Text username and password to be specified in IBM_USERNAME and IBM_PASSWORD environment variables")
|
|
def test_ibm_chinese(self):
|
|
r = sr.Recognizer()
|
|
with sr.AudioFile(self.AUDIO_FILE_ZH) as source: audio = r.record(source)
|
|
self.assertEqual(r.recognize_ibm(audio, username=os.environ["IBM_USERNAME"], password=os.environ["IBM_PASSWORD"], language="zh-CN"), u"砸 自己 的 脚 ")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|