@@ -302,26 +302,26 @@ def test_low_compression(self):
302302 self .assertEqual (openobj .read (1 ), b'2' )
303303
304304 def test_writestr_compression (self ):
305- zipfp = zipfile .ZipFile (TESTFN2 , "w" )
306- zipfp .writestr ("b.txt" , "hello world" , compress_type = self .compression )
307- info = zipfp .getinfo ('b.txt' )
308- self .assertEqual (info .compress_type , self .compression )
305+ with zipfile .ZipFile (TESTFN2 , "w" ) as zipfp :
306+ zipfp .writestr ("b.txt" , "hello world" , compress_type = self .compression )
307+ info = zipfp .getinfo ('b.txt' )
308+ self .assertEqual (info .compress_type , self .compression )
309309
310310 def test_writestr_compresslevel (self ):
311- zipfp = zipfile .ZipFile (TESTFN2 , "w" , compresslevel = 1 )
312- zipfp .writestr ("a.txt" , "hello world" , compress_type = self .compression )
313- zipfp .writestr ("b.txt" , "hello world" , compress_type = self .compression ,
314- compresslevel = 2 )
311+ with zipfile .ZipFile (TESTFN2 , "w" , compresslevel = 1 ) as zipfp :
312+ zipfp .writestr ("a.txt" , "hello world" , compress_type = self .compression )
313+ zipfp .writestr ("b.txt" , "hello world" , compress_type = self .compression ,
314+ compresslevel = 2 )
315315
316- # Compression level follows the constructor.
317- a_info = zipfp .getinfo ('a.txt' )
318- self .assertEqual (a_info .compress_type , self .compression )
319- self .assertEqual (a_info .compress_level , 1 )
316+ # Compression level follows the constructor.
317+ a_info = zipfp .getinfo ('a.txt' )
318+ self .assertEqual (a_info .compress_type , self .compression )
319+ self .assertEqual (a_info .compress_level , 1 )
320320
321- # Compression level is overridden.
322- b_info = zipfp .getinfo ('b.txt' )
323- self .assertEqual (b_info .compress_type , self .compression )
324- self .assertEqual (b_info ._compresslevel , 2 )
321+ # Compression level is overridden.
322+ b_info = zipfp .getinfo ('b.txt' )
323+ self .assertEqual (b_info .compress_type , self .compression )
324+ self .assertEqual (b_info ._compresslevel , 2 )
325325
326326 def test_read_return_size (self ):
327327 # Issue #9837: ZipExtFile.read() shouldn't return more bytes
@@ -2255,13 +2255,15 @@ def test_empty_zipfile(self):
22552255 zipf = zipfile .ZipFile (TESTFN , mode = "r" )
22562256 except zipfile .BadZipFile :
22572257 self .fail ("Unable to create empty ZIP file in 'w' mode" )
2258+ zipf .close ()
22582259
22592260 zipf = zipfile .ZipFile (TESTFN , mode = "a" )
22602261 zipf .close ()
22612262 try :
22622263 zipf = zipfile .ZipFile (TESTFN , mode = "r" )
22632264 except :
22642265 self .fail ("Unable to create empty ZIP file in 'a' mode" )
2266+ zipf .close ()
22652267
22662268 def test_open_empty_file (self ):
22672269 # Issue 1710703: Check that opening a file with less than 22 bytes
0 commit comments