@@ -312,26 +312,26 @@ def test_low_compression(self):
312312 self .assertEqual (openobj .read (1 ), b'2' )
313313
314314 def test_writestr_compression (self ):
315- zipfp = zipfile .ZipFile (TESTFN2 , "w" )
316- zipfp .writestr ("b.txt" , "hello world" , compress_type = self .compression )
317- info = zipfp .getinfo ('b.txt' )
318- self .assertEqual (info .compress_type , self .compression )
315+ with zipfile .ZipFile (TESTFN2 , "w" ) as zipfp :
316+ zipfp .writestr ("b.txt" , "hello world" , compress_type = self .compression )
317+ info = zipfp .getinfo ('b.txt' )
318+ self .assertEqual (info .compress_type , self .compression )
319319
320320 def test_writestr_compresslevel (self ):
321- zipfp = zipfile .ZipFile (TESTFN2 , "w" , compresslevel = 1 )
322- zipfp .writestr ("a.txt" , "hello world" , compress_type = self .compression )
323- zipfp .writestr ("b.txt" , "hello world" , compress_type = self .compression ,
324- compresslevel = 2 )
321+ with zipfile .ZipFile (TESTFN2 , "w" , compresslevel = 1 ) as zipfp :
322+ zipfp .writestr ("a.txt" , "hello world" , compress_type = self .compression )
323+ zipfp .writestr ("b.txt" , "hello world" , compress_type = self .compression ,
324+ compresslevel = 2 )
325325
326- # Compression level follows the constructor.
327- a_info = zipfp .getinfo ('a.txt' )
328- self .assertEqual (a_info .compress_type , self .compression )
329- self .assertEqual (a_info .compress_level , 1 )
326+ # Compression level follows the constructor.
327+ a_info = zipfp .getinfo ('a.txt' )
328+ self .assertEqual (a_info .compress_type , self .compression )
329+ self .assertEqual (a_info .compress_level , 1 )
330330
331- # Compression level is overridden.
332- b_info = zipfp .getinfo ('b.txt' )
333- self .assertEqual (b_info .compress_type , self .compression )
334- self .assertEqual (b_info ._compresslevel , 2 )
331+ # Compression level is overridden.
332+ b_info = zipfp .getinfo ('b.txt' )
333+ self .assertEqual (b_info .compress_type , self .compression )
334+ self .assertEqual (b_info ._compresslevel , 2 )
335335
336336 def test_read_return_size (self ):
337337 # Issue #9837: ZipExtFile.read() shouldn't return more bytes
@@ -2330,13 +2330,15 @@ def test_empty_zipfile(self):
23302330 zipf = zipfile .ZipFile (TESTFN , mode = "r" )
23312331 except zipfile .BadZipFile :
23322332 self .fail ("Unable to create empty ZIP file in 'w' mode" )
2333+ zipf .close ()
23332334
23342335 zipf = zipfile .ZipFile (TESTFN , mode = "a" )
23352336 zipf .close ()
23362337 try :
23372338 zipf = zipfile .ZipFile (TESTFN , mode = "r" )
23382339 except :
23392340 self .fail ("Unable to create empty ZIP file in 'a' mode" )
2341+ zipf .close ()
23402342
23412343 def test_open_empty_file (self ):
23422344 # Issue 1710703: Check that opening a file with less than 22 bytes
0 commit comments