refactor: a better way to skip these tests

1 files changed, 1 insertions, 5 deletions

diff --git a/tests/test_misc.py b/tests/test_misc.py
index 85b3d8f8..dad542ac 100644
--- a/tests/test_misc.py
+++ b/tests/test_misc.py
@@ -76,16 +76,12 @@ class RemoveFileTest(CoverageTest):
             file_be_gone(".")
 
 
+@pytest.mark.skipif(not USE_CONTRACTS, reason="Contracts are disabled, can't test them")
 class ContractTest(CoverageTest):
     """Tests of our contract decorators."""
 
     run_in_temp_dir = False
 
-    def setUp(self):
-        super(ContractTest, self).setUp()
-        if not USE_CONTRACTS:
-            self.skipTest("Contracts are disabled")
-
     def test_bytes(self):
         @contract(text='bytes|None')
         def need_bytes(text=None):