100 lines
4.5 KiB
Python
100 lines
4.5 KiB
Python
import stat
|
|
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from mycroft.tts.mimic_tts import (Mimic, download_subscriber_voices,
|
|
get_mimic_binary,
|
|
get_subscriber_voices)
|
|
|
|
|
|
device_instance_mock = mock.Mock(name='device_api_instance')
|
|
device_instance_mock.is_subscriber = False
|
|
|
|
subscribed_device = mock.Mock(name='subscriber_device')
|
|
subscribed_device.is_subscribed = True
|
|
subscribed_device.get_subscriber_voice_url.return_value = 'https://trinity'
|
|
|
|
|
|
@mock.patch('mycroft.tts.mimic_tts.DeviceApi')
|
|
@mock.patch('mycroft.tts.tts.PlaybackThread')
|
|
class TestMimic(unittest.TestCase):
|
|
@mock.patch('mycroft.tts.mimic_tts.subprocess')
|
|
def test_get_tts(self, mock_subprocess, _, mock_device_api):
|
|
mock_device_api.return_value = device_instance_mock
|
|
mock_subprocess.check_output().decode.return_value = 's:1 pau:2'
|
|
mock_subprocess.check_output.reset_mock()
|
|
m = Mimic('en-US', {})
|
|
wav, phonemes = m.get_tts('hello', 'abc.wav')
|
|
mock_subprocess.check_output.assert_called_once_with(
|
|
m.args + ['-o', 'abc.wav', '-t', 'hello'])
|
|
self.assertEqual(phonemes, [['s', '1'], ['pau', '2']])
|
|
|
|
def test_viseme(self, _, mock_device_api):
|
|
mock_device_api.return_value = device_instance_mock
|
|
m = Mimic('en-US', {})
|
|
phoneme_list = (['pau', 0.206], ['m', 0.287], ['ah', 0.401],
|
|
['ch', 0.513], ['dh', 0.578], ['iy', 0.699],
|
|
['s', 0.835], ['ey', 1.013], ['m', 1.118],
|
|
['w', 1.213], ['ey', 1.345], ['dh', 1.415],
|
|
['ae', 1.491], ['t', 1.539], ['b', 1.616],
|
|
['r', 1.671], ['ih', 1.744], ['k', 1.819],
|
|
['s', 1.923], ['d', 1.978], ['ow', 2.118],
|
|
['n', 2.206], ['t', 2.301], ['pau', 2.408])
|
|
|
|
vis = m.viseme(phoneme_list)
|
|
self.assertEqual(vis,
|
|
[('4', 0.206), ('4', 0.287), ('0', 0.401),
|
|
('3', 0.513), ('3', 0.578), ('0', 0.699),
|
|
('3', 0.835), ('0', 1.013), ('4', 1.118),
|
|
('2', 1.213), ('0', 1.345), ('3', 1.415),
|
|
('0', 1.491), ('3', 1.539), ('4', 1.616),
|
|
('2', 1.671), ('0', 1.744), ('3', 1.819),
|
|
('3', 1.923), ('3', 1.978), ('2', 2.118),
|
|
('3', 2.206), ('3', 2.301), ('4', 2.408)])
|
|
|
|
@mock.patch('mycroft.tts.mimic_tts.Thread')
|
|
def test_subscriber(self, mock_thread, _, mock_device_api):
|
|
mock_device_api.return_value = subscribed_device
|
|
default_mimic = get_mimic_binary()
|
|
trinity_mimic = get_subscriber_voices()['trinity']
|
|
m = Mimic('en-US', {'voice': 'trinity'})
|
|
mock_thread.assert_called_with(target=download_subscriber_voices,
|
|
args=['trinity'])
|
|
self.assertTrue(m.is_subscriber)
|
|
self.assertEqual(m.args,
|
|
[default_mimic, '-voice', 'ap', '-psdur', '-ssml'])
|
|
with mock.patch('mycroft.tts.mimic_tts.exists') as mock_exists:
|
|
mock_exists.return_value = True
|
|
self.assertEqual(m.args, [trinity_mimic, '-voice',
|
|
'trinity', '-psdur', '-ssml'])
|
|
|
|
@mock.patch('mycroft.tts.mimic_tts.sleep')
|
|
@mock.patch('mycroft.tts.mimic_tts.download')
|
|
def test_download(self, mock_download, mock_sleep, _, mock_device_api):
|
|
mock_device_api.return_value = subscribed_device
|
|
dl = mock.Mock()
|
|
dl.done = False
|
|
|
|
def sleep_sideeffect(_):
|
|
"""After one sleep call the download should be considered done."""
|
|
nonlocal dl
|
|
dl.done = True
|
|
|
|
mock_sleep.side_effect = sleep_sideeffect
|
|
mock_download.return_value = dl
|
|
|
|
download_subscriber_voices('trinity')
|
|
self.assertEqual(mock_download.call_args[0][:2],
|
|
('https://trinity', '/opt/mycroft/voices/mimic_tn'))
|
|
make_executable = mock_download.call_args[0][2]
|
|
|
|
# Check that the excutable flag is set to the downloaded file
|
|
with mock.patch('mycroft.tts.mimic_tts.os.chmod') as mock_chmod:
|
|
with mock.patch('mycroft.tts.mimic_tts.os.stat') as mock_stat:
|
|
st_mock = mock.Mock()
|
|
mock_stat.return_value = st_mock
|
|
st_mock.st_mode = 0
|
|
make_executable('/test')
|
|
mock_chmod.assert_called_with('/test', stat.S_IEXEC)
|