mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-24 20:19:22 +01:00
add maoe unit test
This commit is contained in:
@@ -1,5 +1,4 @@
|
||||
# Copyright (c) Alibaba, Inc. and its affiliates.
|
||||
import os
|
||||
import unittest
|
||||
|
||||
from modelscope.pipelines import pipeline
|
||||
@@ -11,7 +10,11 @@ from modelscope.utils.test_utils import test_level
|
||||
class NamedEntityRecognitionTest(unittest.TestCase, DemoCompatibilityCheck):
|
||||
|
||||
def setUp(self):
|
||||
os.system('pip install adaseq>=0.5.0')
|
||||
import subprocess
|
||||
result = subprocess.run(
|
||||
['pip', 'install', 'adaseq>=0.6.2', '--no-deps'],
|
||||
stdout=subprocess.PIPE)
|
||||
print(result.stdout.decode('utf-8'))
|
||||
|
||||
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
|
||||
def test_run_span_based_ner_pipeline(self):
|
||||
@@ -23,3 +26,12 @@ class NamedEntityRecognitionTest(unittest.TestCase, DemoCompatibilityCheck):
|
||||
'1、可测量目标: 1周内胸闷缓解。2、下一步诊疗措施:1.心内科护理常规,一级护理,低盐低脂饮食,留陪客。'
|
||||
'2.予“阿司匹林肠溶片”抗血小板聚集,“呋塞米、螺内酯”利尿减轻心前负荷,“瑞舒伐他汀”调脂稳定斑块,“厄贝沙坦片片”降血压抗心机重构'
|
||||
))
|
||||
|
||||
def test_maoe_pipelines(self):
|
||||
pipeline_ins = pipeline(
|
||||
Tasks.named_entity_recognition,
|
||||
'damo/nlp_maoe_named-entity-recognition_chinese-base-general')
|
||||
print(
|
||||
pipeline_ins(
|
||||
'刘培强,男,生理年龄40岁(因为在太空中进入休眠状态),实际年龄52岁,领航员国际空间站中的中国航天员,机械工程专家,军人,军衔中校。'
|
||||
))
|
||||
|
||||
Reference in New Issue
Block a user