diff --git a/tests/pipelines/test_fill_mask.py b/tests/pipelines/test_fill_mask.py index 87f71b31..49c5dc8a 100644 --- a/tests/pipelines/test_fill_mask.py +++ b/tests/pipelines/test_fill_mask.py @@ -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'