first commit
This commit is contained in:
1
venv/lib/python3.11/site-packages/tests/__init__.py
Normal file
1
venv/lib/python3.11/site-packages/tests/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
# placeholder file to make this folder a module - this allows tests in this folder to be discovered by `python -m unittest discover`
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
146
venv/lib/python3.11/site-packages/tests/test_audio.py
Normal file
146
venv/lib/python3.11/site-packages/tests/test_audio.py
Normal file
@@ -0,0 +1,146 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import unittest
|
||||
from os import path
|
||||
|
||||
import speech_recognition as sr
|
||||
|
||||
|
||||
class TestAudioFile(unittest.TestCase):
|
||||
def assertSimilar(self, bytes_1, bytes_2):
|
||||
for i, (byte_1, byte_2) in enumerate(zip(bytes_1, bytes_2)):
|
||||
if abs(byte_1 - byte_2) > 2:
|
||||
raise AssertionError("{} is really different from {} at index {}".format(bytes_1, bytes_2, i))
|
||||
|
||||
def test_get_segment(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-mono-32-bit-44100Hz.wav")) as source: audio = r.record(source)
|
||||
self.assertEqual(audio.get_raw_data(), audio.get_segment().get_raw_data())
|
||||
self.assertEqual(audio.get_raw_data()[8:], audio.get_segment(0.022675738 * 2).get_raw_data())
|
||||
self.assertEqual(audio.get_raw_data()[:16], audio.get_segment(None, 0.022675738 * 4).get_raw_data())
|
||||
self.assertEqual(audio.get_raw_data()[8:16], audio.get_segment(0.022675738 * 2, 0.022675738 * 4).get_raw_data())
|
||||
|
||||
def test_wav_mono_8_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-mono-8-bit-44100Hz.wav")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 1)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\xff\x00\xff\x00\xff\xff\x00\xff\x00\xff\x00\xff\x00\x00\xff\x00\x00\xff\x00\xff\x00\xff\x00\xff\x00\xff\x00\xff\x00\xff\xff")
|
||||
|
||||
def test_wav_mono_16_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-mono-16-bit-44100Hz.wav")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 2)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\xff\xff\x01\x00\xff\xff\x00\x00\x01\x00\xfe\xff\x01\x00\xfe\xff\x04\x00\xfc\xff\x04\x00\xfe\xff\xff\xff\x03\x00\xfe\xff")
|
||||
|
||||
def test_wav_mono_24_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-mono-24-bit-44100Hz.wav")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
if audio.sample_width == 3:
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\xff\xff\x00\x01\x00\x00\xff\xff\x00\x00\x00\x00\x01\x00\x00\xfe\xff\x00\x01\x00\x00\xfe\xff\x00\x04\x00\x00\xfb")
|
||||
else:
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x01\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\xfe\xff\x00\x00\x01\x00")
|
||||
|
||||
def test_wav_mono_32_bit(self):
|
||||
r = sr.Recognizer()
|
||||
audio_file_path = path.join(path.dirname(path.realpath(__file__)), "audio-mono-32-bit-44100Hz.wav")
|
||||
with sr.AudioFile(audio_file_path) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 4)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x01\x00\x00\x00\xff\xff\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\xfe\xff\x00\x00\x01\x00")
|
||||
|
||||
def test_wav_stereo_8_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-stereo-8-bit-44100Hz.wav")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 1)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\xff\x00\xff\x00\x00\xff\x7f\x7f\x00\xff\x00\xff\x00\x00\xff\x00\x7f\x7f\x7f\x00\x00\xff\x00\xff\x00\xff\x00\x7f\x7f\x7f\x7f")
|
||||
|
||||
def test_wav_stereo_16_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-stereo-16-bit-44100Hz.wav")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 2)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x02\x00\xfb\xff\x04\x00\xfe\xff\xfe\xff\x07\x00\xf6\xff\x07\x00\xf9\xff\t\x00\xf5\xff\x0c\x00\xf8\xff\x02\x00\x04\x00\xfa\xff")
|
||||
|
||||
def test_wav_stereo_24_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-stereo-24-bit-44100Hz.wav")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
if audio.sample_width == 3:
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\xfe\xff\x00\x02\x00\x00\xfe\xff\x00\x00\x00\x00\x02\x00\x00\xfc\xff\x00\x02\x00\x00\xfc\xff\x00\x08\x00\x00\xf6")
|
||||
else:
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\x00\x00\xfe\xff\x00\x00\x02\x00\x00\x00\xfe\xff\x00\x00\x00\x00\x00\x00\x02\x00\x00\x00\xfc\xff\x00\x00\x02\x00")
|
||||
|
||||
def test_wav_stereo_32_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-stereo-32-bit-44100Hz.wav")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 4)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\x00\x00\xfe\xff\x00\x00\x02\x00\x00\x00\xfe\xff\x00\x00\x00\x00\x00\x00\x02\x00\x00\x00\xfc\xff\x00\x00\x02\x00")
|
||||
|
||||
def test_aiff_mono_16_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-mono-16-bit-44100Hz.aiff")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 2)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\xff\xff\x01\x00\xff\xff\x01\x00\xfe\xff\x02\x00\xfd\xff\x04\x00\xfc\xff\x03\x00\x00\x00\xfe\xff\x03\x00\xfd\xff")
|
||||
|
||||
def test_aiff_stereo_16_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-stereo-16-bit-44100Hz.aiff")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 2)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\xfe\xff\x02\x00\xfe\xff\xff\xff\x04\x00\xfa\xff\x04\x00\xfa\xff\t\x00\xf6\xff\n\x00\xfa\xff\xff\xff\x08\x00\xf5\xff")
|
||||
|
||||
def test_flac_mono_16_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-mono-16-bit-44100Hz.flac")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 2)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\xff\xff\x01\x00\xff\xff\x00\x00\x01\x00\xfe\xff\x02\x00\xfc\xff\x06\x00\xf9\xff\x06\x00\xfe\xff\xfe\xff\x05\x00\xfa\xff")
|
||||
|
||||
def test_flac_mono_24_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-mono-24-bit-44100Hz.flac")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
if audio.sample_width == 3:
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\xff\xfe\xff\x02\x01\x00\xfd\xfe\xff\x04\x00\x00\xfc\x00\x00\x04\xfe\xff\xfb\x00\x00\x05\xfe\xff\xfc\x03\x00\x04\xfb")
|
||||
else:
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\x00\xff\xfe\xff\x00\x02\x01\x00\x00\xfd\xfe\xff\x00\x04\x00\x00\x00\xfc\x00\x00\x00\x04\xfe\xff\x00\xfb\x00\x00")
|
||||
|
||||
def test_flac_stereo_16_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-stereo-16-bit-44100Hz.flac")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
self.assertEqual(audio.sample_width, 2)
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\xff\xff\xff\xff\x02\x00\xfe\xff\x00\x00\x01\x00\xfd\xff\x01\x00\xff\xff\x04\x00\xfa\xff\x05\x00\xff\xff\xfd\xff\x08\x00\xf6\xff")
|
||||
|
||||
def test_flac_stereo_24_bit(self):
|
||||
r = sr.Recognizer()
|
||||
with sr.AudioFile(path.join(path.dirname(path.realpath(__file__)), "audio-stereo-24-bit-44100Hz.flac")) as source: audio = r.record(source)
|
||||
self.assertIsInstance(audio, sr.AudioData)
|
||||
self.assertEqual(audio.sample_rate, 44100)
|
||||
if audio.sample_width == 3:
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\xfe\xff\x00\x02\x00\x00\xfe\xff\x00\x00\x00\xff\x01\x00\x02\xfc\xff\xfe\x01\x00\x02\xfc\xff\xfe\x07\x00\x01\xf6")
|
||||
else:
|
||||
self.assertSimilar(audio.get_raw_data()[:32], b"\x00\x00\x00\x00\x00\x00\xfe\xff\x00\x00\x02\x00\x00\x00\xfe\xff\x00\x00\x00\x00\x00\xff\x01\x00\x00\x02\xfc\xff\x00\xfe\x01\x00")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
88
venv/lib/python3.11/site-packages/tests/test_recognition.py
Normal file
88
venv/lib/python3.11/site-packages/tests/test_recognition.py
Normal file
@@ -0,0 +1,88 @@
|
||||
#!/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()
|
||||
@@ -0,0 +1,34 @@
|
||||
#!/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()
|
||||
Reference in New Issue
Block a user