From d5ee8aa66d4ea6c42f463a9db3a6f198c2ecc3fc Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Mon, 28 Nov 2022 13:50:28 +0800 Subject: [PATCH] move long-running test to level 2 --- tests/pipelines/test_extractive_summarization.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/test_extractive_summarization.py b/tests/pipelines/test_extractive_summarization.py index 8bf28fd2..26ac508c 100644 --- a/tests/pipelines/test_extractive_summarization.py +++ b/tests/pipelines/test_extractive_summarization.py @@ -28,7 +28,7 @@ class ExtractiveSummarizationTest(unittest.TestCase, DemoCompatibilityCheck): result = p(documents=documents) return result - @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_doc(self): logger.info( 'Run doc extractive summarization (PoNet) with one document ...') @@ -37,7 +37,7 @@ class ExtractiveSummarizationTest(unittest.TestCase, DemoCompatibilityCheck): model_id=self.ponet_doc_model_id, documents=self.sentences) print(result[OutputKeys.TEXT]) - @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_topic(self): logger.info( 'Run topic extractive summarization (PoNet) with one document ...')