Remove add_compress_level_flag helper in the test of gzip

Author: matrixise

Lib/test/test_gzip.py

@@ -687,19 +687,6 @@ def wrapper(*args, **kwargs):
     return decorator
 
 
-def add_compress_level_flag(*compress_levels):
-    def decorator(function):
-        @functools.wraps(function)
-        def wrapper(*args, **kwargs):
-            self = args[0]
-            for compress_level in compress_levels:
-                with self.subTest(compress_level=compress_level):
-                    kwargs.update(compress_level=compress_level)
-                    function(*args, **kwargs)
-        return wrapper
-    return decorator
-
-
 class TestCommandLine(unittest.TestCase):
     data = b'This is a simple test with gzip'
 
@@ -763,22 +750,24 @@ def test_compress_infile_outfile(self):
         self.assertEqual(out, b'')
         self.assertEqual(err, b'')
 
-    @add_compress_level_flag('--fast', '--best')
     @create_and_remove_directory(TEMPDIR)
-    def test_compress_infile_outfile(self, compress_level):
-        local_testgzip = os.path.join(TEMPDIR, 'testgzip')
-        gzipname = local_testgzip + '.gz'
-        self.assertFalse(os.path.exists(gzipname))
-
-        with open(local_testgzip, 'wb') as fp:
-            fp.write(self.data)
-
-        rc, out, err = assert_python_ok('-m', 'gzip', compress_level, local_testgzip)
-
-        self.assertTrue(os.path.exists(gzipname))
-        self.assertEqual(rc, 0)
-        self.assertEqual(out, b'')
-        self.assertEqual(err, b'')
+    def test_compress_infile_outfile(self):
+        for compress_level in ('--fast', '--best'):
+            with self.subTest(compress_level=compress_level):
+                local_testgzip = os.path.join(TEMPDIR, 'testgzip')
+                gzipname = local_testgzip + '.gz'
+                self.assertFalse(os.path.exists(gzipname))
+
+                with open(local_testgzip, 'wb') as fp:
+                    fp.write(self.data)
+
+                rc, out, err = assert_python_ok('-m', 'gzip', compress_level, local_testgzip)
+
+                self.assertTrue(os.path.exists(gzipname))
+                self.assertEqual(out, b'')
+                self.assertEqual(err, b'')
+                os.remove(gzipname)
+                self.assertFalse(os.path.exists(gzipname))
 
     def test_compress_fast_best_are_exclusive(self):
         rc, out, err = assert_python_failure('-m', 'gzip', '--fast', '--best')