Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933] add en2ru and ru2en pipeline ut v4

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11549835

    * add en2ru and ru2en pipeline ut v4
This commit is contained in:
xiangpeng.wxp
2023-02-06 05:26:55 +00:00
parent 8fb2632e26
commit cff63fb79b
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/pipelines/test_csanmt_translation.py
View File

@@ -56,6 +56,13 @@ class TranslationTest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(self.task, model=model_id)
print(pipeline_ins(input=inputs))
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_run_with_model_name_for_en2ru(self):
model_id = 'damo/nlp_csanmt_translation_en2ru_base'
inputs = 'When I was in my 20s, I saw my very first psychotherapy client.'
pipeline_ins = pipeline(self.task, model=model_id)
print(pipeline_ins(input=inputs))
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_run_with_model_name_for_fr2en(self):
model_id = 'damo/nlp_csanmt_translation_fr2en'
@@ -70,6 +77,13 @@ class TranslationTest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(self.task, model=model_id)
print(pipeline_ins(input=inputs))
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_run_with_model_name_for_ru2en(self):
model_id = 'damo/nlp_csanmt_translation_ru2en_base'
inputs = 'Это всего лишь пример.'
pipeline_ins = pipeline(self.task, model=model_id)
print(pipeline_ins(input=inputs))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
def test_run_with_default_model(self):
inputs = '声明补充说，沃伦的同事都深感震惊，并且希望他能够投案自首。'