remove uttest

This commit is contained in:
shuaigezhu
2022-11-25 11:55:53 +08:00
parent 1ab8a1f764
commit 65adde14d8

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()