@@ -192,16 +192,30 @@ def test_shebang(self):
192
192
Find Python files that don't end with `.py`, but contain a Python
193
193
shebang.
194
194
"""
195
- apath = os .path .join (self .tempdir , 'a' )
196
- fd = open (apath , 'w' )
197
- fd .write ('#!/usr/bin/env python\n ' )
198
- fd .close ()
195
+ python = os .path .join (self .tempdir , 'a' )
196
+ with open (python , 'w' ) as fd :
197
+ fd .write ('#!/usr/bin/env python\n ' )
199
198
200
199
self .makeEmptyFile ('b' )
201
200
201
+ with open (os .path .join (self .tempdir , 'c' ), 'w' ) as fd :
202
+ fd .write ('hello\n world\n ' )
203
+
204
+ python2 = os .path .join (self .tempdir , 'd' )
205
+ with open (python2 , 'w' ) as fd :
206
+ fd .write ('#!/usr/bin/env python2\n ' )
207
+
208
+ python3 = os .path .join (self .tempdir , 'e' )
209
+ with open (python3 , 'w' ) as fd :
210
+ fd .write ('#!/usr/bin/env python3\n ' )
211
+
212
+ pythonw = os .path .join (self .tempdir , 'f' )
213
+ with open (pythonw , 'w' ) as fd :
214
+ fd .write ('#!/usr/bin/env pythonw\n ' )
215
+
202
216
self .assertEqual (
203
- list (iterSourceCode ([self .tempdir ])),
204
- list ([ apath ]))
217
+ sorted (iterSourceCode ([self .tempdir ])),
218
+ sorted ([ python , python2 , python3 , pythonw ]))
205
219
206
220
def test_multipleDirectories (self ):
207
221
"""
0 commit comments