| @@ -51,7 +51,7 @@ class ImageDenoiseTrainerTest(unittest.TestCase): | |||||
| shutil.rmtree(self.tmp_dir, ignore_errors=True) | shutil.rmtree(self.tmp_dir, ignore_errors=True) | ||||
| super().tearDown() | super().tearDown() | ||||
| @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') | |||||
| @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') | |||||
| def test_trainer(self): | def test_trainer(self): | ||||
| kwargs = dict( | kwargs = dict( | ||||
| model=self.model_id, | model=self.model_id, | ||||
| @@ -65,7 +65,7 @@ class ImageDenoiseTrainerTest(unittest.TestCase): | |||||
| for i in range(2): | for i in range(2): | ||||
| self.assertIn(f'epoch_{i+1}.pth', results_files) | self.assertIn(f'epoch_{i+1}.pth', results_files) | ||||
| @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') | |||||
| @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') | |||||
| def test_trainer_with_model_and_args(self): | def test_trainer_with_model_and_args(self): | ||||
| model = NAFNetForImageDenoise.from_pretrained(self.cache_path) | model = NAFNetForImageDenoise.from_pretrained(self.cache_path) | ||||
| kwargs = dict( | kwargs = dict( | ||||