def test_load_stt_plugin(self): # Test getting default stt ol = OrderListener() stt1 = Stt(name="default-stt", parameters=dict()) stt2 = Stt(name="second-stt", parameters=dict()) stt3 = Stt(name="third-stt", parameters=dict()) resources = Resources(stt_folder="/tmp") ol.settings = mock.MagicMock(default_stt_name="default-stt", stts=[stt1, stt2, stt3], resources=resources) callback = mock.MagicMock() ol.callback = callback with mock.patch("kalliope.core.Utils.get_dynamic_class_instantiation" ) as mock_get_dynamic_class_instantiation: mock_get_dynamic_class_instantiation.return_value = 'class_instance' self.assertEqual(ol.load_stt_plugin(), "class_instance", "Fail getting the proper value") mock_get_dynamic_class_instantiation.assert_called_once_with( package_name="stt", module_name="Default-stt", parameters={ 'callback': callback, 'audio_file_path': None }, resources_dir="/tmp")
def test_load_stt_plugin(self): # Test getting default stt ol = OrderListener() stt1 = Stt(name="default-stt", parameters=dict()) stt2 = Stt(name="second-stt", parameters=dict()) stt3 = Stt(name="third-stt", parameters=dict()) resources = Resources(stt_folder="/tmp") ol.settings = mock.MagicMock(default_stt_name="default-stt", stts=[stt1,stt2,stt3], resources=resources) callback = mock.MagicMock() ol.callback = callback with mock.patch("kalliope.core.Utils.get_dynamic_class_instantiation") as mock_get_dynamic_class_instantiation: mock_get_dynamic_class_instantiation.return_value = 'class_instance' self.assertEqual(ol.load_stt_plugin(), "class_instance", "Fail getting the proper value") mock_get_dynamic_class_instantiation.assert_called_once_with(package_name= "stt", module_name= "Default-stt", parameters={'callback' : callback}, resources_dir= "/tmp")
def show_stt_test_menu(self): """ Show the list of available STT. Clicking on a STT will load the engine to catch the user audio and return a text """ # we get STT from settings stt_list = self.settings.stts logger.debug("Loaded stt list: %s" % str(stt_list)) choices = self._get_choices_tuple_from_list(stt_list) code, tag = self.d.menu("Select the STT to test:", choices=choices) # go back to the main menu if we choose "cancel" if code == self.d.CANCEL: self.show_main_menu() # if ok, call the target TTS engine and catch audio if code == self.d.OK: self.d.infobox("Please talk now") # the callback funtion will print the translated audio into text on the screen order_listener = OrderListener(callback=self.callback_stt, stt=str(tag)) order_listener.load_stt_plugin()