shuaigezhu
3 years ago
1 changed files with 0 additions and 31 deletions
Unified View
Diff Options
+ 0
- 31
tests/pipelines/test_CodeGeeX_code_translation.py
View File
| @@ -1,31 +0,0 @@ | |||||
| # Copyright (c) Alibaba, Inc. and its affiliates. | |||||
| import os | |||||
| import unittest | |||||
| from modelscope.pipelines import pipeline | |||||
| from modelscope.utils.constant import Tasks | |||||
| from modelscope.utils.demo_utils import DemoCompatibilityCheck | |||||
| from modelscope.utils.test_utils import test_level | |||||
| class CodeGeeXCodeTranslationTest(unittest.TestCase, DemoCompatibilityCheck): | |||||
| def setUp(self) -> None: | |||||
| self.output_dir = 'unittest_output' | |||||
| os.makedirs(self.output_dir, exist_ok=True) | |||||
| @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') | |||||
| def test_run_with_CodeGeeX_with_name(self): | |||||
| model = 'ZhipuAI/CodeGeeX-Code-Translation-13B' | |||||
| pipe = pipeline(task=Tasks.code_translation, model=model) | |||||
| inputs = { | |||||
| 'prompt': 'for i in range(10):\n\tprint(i)\n', | |||||
| 'source language': 'Python', | |||||
| 'target language': 'C++' | |||||
| } | |||||
| result = pipe(inputs) | |||||
| print(result) | |||||
| if __name__ == '__main__': | |||||
| unittest.main() | |||||