|
|
|
@@ -35,7 +35,7 @@ class FillMaskTest(unittest.TestCase): |
|
|
|
'[MASK]. Your [MASK] universe is just a mirror [MASK] of your story.' |
|
|
|
} |
|
|
|
|
|
|
|
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level') |
|
|
|
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level') |
|
|
|
def test_run_by_direct_model_download(self): |
|
|
|
# sbert |
|
|
|
for language in ['zh', 'en']: |
|
|
|
@@ -115,7 +115,7 @@ class FillMaskTest(unittest.TestCase): |
|
|
|
f'\nori_text: {self.ori_texts[language]}\ninput: {self.test_inputs[language]}\npipeline: ' |
|
|
|
f'{pipeline_ins(self.test_inputs[language])}\n') |
|
|
|
|
|
|
|
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level') |
|
|
|
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level') |
|
|
|
def test_run_with_default_model(self): |
|
|
|
pipeline_ins = pipeline(task=Tasks.fill_mask) |
|
|
|
language = 'en' |
|
|
|
|