def test_ibm_stt(self, mock_get, mock_post): import json config = base_config() config.merge({ 'stt': { 'module': 'ibm', 'ibm': { 'credential': { 'token': 'FOOBAR' }, 'url': 'https://test.com' }, }, 'lang': 'en-US' }) mock_get.return_value = config requests_object = MagicMock() requests_object.status_code = 200 requests_object.text = json.dumps({ 'results': [{ 'alternatives': [{ 'confidence': 0.96, 'transcript': 'sample response' }], 'final': True }], 'result_index': 0 }) mock_post.return_value = requests_object audio = MagicMock() audio.sample_rate = 16000 stt = mycroft.stt.IBMSTT() stt.execute(audio) test_url_base = 'https://test.com/v1/recognize' mock_post.assert_called_with(test_url_base, auth=('apikey', 'FOOBAR'), headers={ 'Content-Type': 'audio/x-flac', 'X-Watson-Learning-Opt-Out': 'true' }, data=audio.get_flac_data(), params={ 'model': 'en-US_BroadbandModel', 'profanity_filter': 'false' })
def test_wit_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = {'stt': { 'module': 'wit', 'wit': {'credential': {'token': 'FOOBAR'}}, }, "lang": "en-US" } mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.WITSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_wit.called)
def test_mycroft_stt(self, mock_get): mycroft.stt.STTApi = mock.MagicMock() config = {'stt': { 'module': 'mycroft', 'mycroft': {'uri': 'https://test.com'} }, 'lang': 'en-US' } mock_get.return_value = config stt = mycroft.stt.MycroftSTT() audio = mock.MagicMock() stt.execute(audio, 'en-us') self.assertTrue(mycroft.stt.STTApi.called)
def test_google_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = {'stt': { 'module': 'google', 'google': {'credential': {'token': 'FOOBAR'}}, }, "lang": "en-US" } mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.GoogleSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_google.called)
def test_houndify_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = {'stt': { 'module': 'houndify', 'houndify': {'credential': {'client_id': 'FOO', "client_key": "BAR"}} }, "lang": "en-US" } mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.HoundifySTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_houndify.called)
def test_ibm_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = {'stt': { 'module': 'ibm', 'ibm': {'credential': {'username': '******', "password": '******'}}, }, "lang": "en-US" } mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.IBMSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_ibm.called)
def test_kaldi_stt(self, mock_get, mock_post): mycroft.stt.Recognizer = mock.MagicMock config = { 'stt': { 'module': 'kaldi', 'kaldi': { 'uri': 'https://test.com' }, }, "lang": "en-US" } mock_get.return_value = config kaldiResponse = mock.MagicMock() kaldiResponse.json.return_value = { 'hypotheses': [{ 'utterance': ' [noise] text' }, { 'utterance': ' asdf' }] } mock_post.return_value = kaldiResponse audio = mock.MagicMock() stt = mycroft.stt.KaldiSTT() self.assertEquals(stt.execute(audio), 'text')
def test_bing_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = base_config() config.merge( { 'stt': { 'module': 'bing', 'bing': {'credential': {'token': 'FOOBAR'}}, }, 'lang': 'en-US' }) mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.BingSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_bing.called)
def test_google_cloud_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = {'stt': { 'module': 'google_cloud', 'google_cloud': { 'credential': { 'json': {} } }, }, "lang": "en-US" } mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.GoogleCloudSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_google_cloud.called)
def test_ibm_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = base_config() config.merge( { 'stt': { 'module': 'ibm', 'ibm': { 'credential': {'username': '******', 'password': '******'} }, }, 'lang': 'en-US' }) mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.IBMSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_ibm.called)
def test_houndify_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = { 'stt': { 'module': 'houndify', 'houndify': { 'credential': { 'client_id': 'FOO', "client_key": "BAR" } } }, "lang": "en-US" } mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.HoundifySTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_houndify.called)
def test_google_cloud_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = base_config() config.merge({ 'stt': { 'module': 'google_cloud', 'google_cloud': { 'credential': { 'json': {} } }, }, 'lang': 'en-US' }) mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.GoogleCloudSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_google_cloud.called)
def test_bing_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = base_config() config.merge({ 'stt': { 'module': 'bing', 'bing': { 'credential': { 'token': 'FOOBAR' } }, }, 'lang': 'en-US' }) mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.BingSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_bing.called)
def test_ibm_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = { 'stt': { 'module': 'ibm', 'ibm': { 'credential': { 'username': '******', "password": '******' } }, }, "lang": "en-US" } mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.IBMSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_ibm.called)
def test_ibm_stt(self, mock_get): mycroft.stt.Recognizer = mock.MagicMock config = base_config() config.merge({ 'stt': { 'module': 'ibm', 'ibm': { 'credential': { 'username': '******', 'password': '******' } }, }, 'lang': 'en-US' }) mock_get.return_value = config audio = mock.MagicMock() stt = mycroft.stt.IBMSTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_ibm.called)
def test_houndify_stt(self, mock_get): mycroft.stt.Recognizer = MagicMock config = base_config() config.merge({ 'stt': { 'module': 'houndify', 'houndify': { 'credential': { 'client_id': 'FOO', 'client_key': "BAR" } } }, 'lang': 'en-US' }) mock_get.return_value = config audio = MagicMock() stt = mycroft.stt.HoundifySTT() stt.execute(audio) self.assertTrue(stt.recognizer.recognize_houndify.called)
def test_kaldi_stt(self, mock_get, mock_post): mycroft.stt.Recognizer = mock.MagicMock config = {'stt': { 'module': 'kaldi', 'kaldi': {'uri': 'https://test.com'}, }, "lang": "en-US" } mock_get.return_value = config kaldiResponse = mock.MagicMock() kaldiResponse.json.return_value = { 'hypotheses': [{'utterance': ' [noise] text'}, {'utterance': ' asdf'}] } mock_post.return_value = kaldiResponse audio = mock.MagicMock() stt = mycroft.stt.KaldiSTT() self.assertEquals(stt.execute(audio), 'text')