Spaces:
Running
Running
File size: 1,654 Bytes
85b7206 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import unittest
import numpy as np
from ModelInterfaces import IASRModel, ITranslationModel, ITextToSpeechModel, ITextToPhonemModel
class TestModelInterfaces(unittest.TestCase):
def test_IASRModel(self):
class ASRModel(IASRModel):
def __init__(self):
self.decoder = 1
self.model = 2
def getTranscript(self):
return "transcript"
def getWordLocations(self):
return [(0, 1)]
def processAudio(self, audio):
return "processed audio"
self.assertTrue(issubclass(ASRModel, IASRModel))
asr_model = ASRModel()
assert asr_model.decoder == 1
assert asr_model.model == 2
def test_ITranslationModel_subclasshook(self):
class TranslationModel(ITranslationModel):
def translateSentence(self, sentence):
return "translated sentence"
self.assertTrue(issubclass(TranslationModel, ITranslationModel))
translation_model = TranslationModel()
def test_ITextToSpeechModel_subclasshook(self):
class TextToSpeechModel(ITextToSpeechModel):
def getAudioFromSentence(self, sentence):
return np.array([0.1, 0.2, 0.3])
self.assertTrue(issubclass(TextToSpeechModel, ITextToSpeechModel))
def test_ITextToPhonemModel_subclasshook(self):
class TextToPhonemModel(ITextToPhonemModel):
def convertToPhonem(self, sentence):
return "phonemes"
self.assertTrue(issubclass(TextToPhonemModel, ITextToPhonemModel))
if __name__ == '__main__':
unittest.main() |