@@ -14,13 +14,11 @@ def setUp(self):
1414 self .name_diff = support .TESTFN + '-diff'
1515 data = 'Contents of file go here.\n '
1616 for name in [self .name , self .name_same , self .name_diff ]:
17- output = open (name , 'w' )
18- output .write (data )
19- output .close ()
17+ with open (name , 'w' ) as output :
18+ output .write (data )
2019
21- output = open (self .name_diff , 'a+' )
22- output .write ('An extra line.\n ' )
23- output .close ()
20+ with open (self .name_diff , 'a+' ) as output :
21+ output .write ('An extra line.\n ' )
2422 self .dir = tempfile .gettempdir ()
2523
2624 def tearDown (self ):
@@ -71,13 +69,11 @@ def setUp(self):
7169 fn = 'FiLe' # Verify case-insensitive comparison
7270 else :
7371 fn = 'file'
74- output = open (os .path .join (dir , fn ), 'w' )
75- output .write (data )
76- output .close ()
72+ with open (os .path .join (dir , fn ), 'w' ) as output :
73+ output .write (data )
7774
78- output = open (os .path .join (self .dir_diff , 'file2' ), 'w' )
79- output .write ('An extra file.\n ' )
80- output .close ()
75+ with open (os .path .join (self .dir_diff , 'file2' ), 'w' ) as output :
76+ output .write ('An extra file.\n ' )
8177
8278 def tearDown (self ):
8379 for dir in (self .dir , self .dir_same , self .dir_diff ):
@@ -104,9 +100,8 @@ def test_cmpfiles(self):
104100 "Comparing directory to same fails" )
105101
106102 # Add different file2
107- output = open (os .path .join (self .dir , 'file2' ), 'w' )
108- output .write ('Different contents.\n ' )
109- output .close ()
103+ with open (os .path .join (self .dir , 'file2' ), 'w' ) as output :
104+ output .write ('Different contents.\n ' )
110105
111106 self .assertFalse (filecmp .cmpfiles (self .dir , self .dir_same ,
112107 ['file' , 'file2' ]) ==
@@ -178,9 +173,8 @@ def test_dircmp(self):
178173 self ._assert_report (d .report , expected_report )
179174
180175 # Add different file2
181- output = open (os .path .join (self .dir_diff , 'file2' ), 'w' )
182- output .write ('Different contents.\n ' )
183- output .close ()
176+ with open (os .path .join (self .dir_diff , 'file2' ), 'w' ) as output :
177+ output .write ('Different contents.\n ' )
184178 d = filecmp .dircmp (self .dir , self .dir_diff )
185179 self .assertEqual (d .same_files , ['file' ])
186180 self .assertEqual (d .diff_files , ['file2' ])
0 commit comments