Address code review

python · Nov 2, 2024 · 34c797c · 34c797c
1 parent 97624f6
commit 34c797c
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/Lib/test/test_zlib.py b/Lib/test/test_zlib.py
@@ -505,20 +505,17 @@ def test_flushes(self):
 
         for sync in sync_opt:
             for level in range(10):
-                try:
+                with self.subTest(sync=sync, level=level):
                     obj = zlib.compressobj( level )
                     a = obj.compress( data[:3000] )
                     b = obj.flush( sync )
                     c = obj.compress( data[3000:] )
                     d = obj.flush()
-                except zlib.error:
-                    print("Error for flush mode={}, level={}"
-                          .format(sync, level))
-                    raise
-                self.assertEqual(zlib.decompress(b''.join([a,b,c,d])),
-                                 data, ("Decompress failed: flush "
-                                        "mode=%i, level=%i") % (sync, level))
-                del obj
+
+                    self.assertEqual(zlib.decompress(b''.join([a,b,c,d])),
+                    data, ("Decompress failed: flush "
+                    "mode=%i, level=%i") % (sync, level))
+                    del obj
 
     @unittest.skipUnless(hasattr(zlib, 'Z_SYNC_FLUSH'),
                          'requires zlib.Z_SYNC_FLUSH')