From a8f803fc006b2016592051e2af15b2993631dce0 Mon Sep 17 00:00:00 2001 From: speech_asr Date: Fri, 10 Mar 2023 17:54:10 +0800 Subject: [PATCH] update unittest --- tests/test_asr_inference_pipeline.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/tests/test_asr_inference_pipeline.py b/tests/test_asr_inference_pipeline.py index 0afdc9173..6257d114c 100644 --- a/tests/test_asr_inference_pipeline.py +++ b/tests/test_asr_inference_pipeline.py @@ -449,6 +449,24 @@ class TestUniasrInferencePipelines(unittest.TestCase): param_dict={"decoding_model": "normal"}) logger.info("asr inference result: {0}".format(rec_result)) + def test_uniasr_2pass_zhcn_16k_common_vocab8358_offline(self): + inference_pipeline = pipeline( + task=Tasks.auto_speech_recognition, + model='damo/speech_UniASR_asr_2pass-zh-cn-16k-common-vocab8358-tensorflow1-offline') + rec_result = inference_pipeline( + audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav', + param_dict={"decoding_model": "offline"}) + logger.info("asr inference result: {0}".format(rec_result)) + + def test_uniasr_2pass_zhcn_16k_common_vocab8358_online(self): + inference_pipeline = pipeline( + task=Tasks.auto_speech_recognition, + model='damo/speech_UniASR_asr_2pass-zh-cn-16k-common-vocab8358-tensorflow1-online') + rec_result = inference_pipeline( + audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav', + param_dict={"decoding_model": "normal"}) + logger.info("asr inference result: {0}".format(rec_result)) + if __name__ == '__main__':