1- # -*- coding: utf-8 -*-
2-
31import contextlib
42import inspect
53import os
@@ -22,7 +20,7 @@ def test_constants():
2220 assert EX_DATAERR == 65
2321
2422
25- class MainWrapper ( object ) :
23+ class MainWrapper :
2624 """Compatibility wrapper for when we used to return the count."""
2725
2826 def main (self , * args , count = True , std = False , ** kwargs ):
@@ -174,7 +172,7 @@ def test_interactivity(tmpdir, capsys):
174172 with FakeStdin ('0\n ' ): # blank input -> nothing
175173 assert cs .main ('-w' , '-i' , '3' , f .name ) == 0
176174 assert cs .main (f .name ) == 0
177- with open (f .name , 'r' ) as f_read :
175+ with open (f .name ) as f_read :
178176 assert f_read .read () == 'awkward\n '
179177 with open (f .name , 'w' ) as f :
180178 f .write ('ackward\n ' )
@@ -184,7 +182,7 @@ def test_interactivity(tmpdir, capsys):
184182 assert code == 0
185183 assert 'a valid option' in stdout
186184 assert cs .main (f .name ) == 0
187- with open (f .name , 'r' ) as f :
185+ with open (f .name ) as f :
188186 assert f .read () == 'backward\n '
189187 finally :
190188 os .remove (f .name )
@@ -249,11 +247,11 @@ def test_exclude_file(tmpdir, capsys):
249247 """Test exclude file functionality."""
250248 d = str (tmpdir )
251249 with open (op .join (d , 'bad.txt' ), 'wb' ) as f :
252- f .write ('1 abandonned 1\n 2 abandonned 2\n ' . encode ( 'utf-8' ) )
250+ f .write (b '1 abandonned 1\n 2 abandonned 2\n ' )
253251 bad_name = f .name
254252 assert cs .main (bad_name ) == 2
255253 with open (op .join (d , 'tmp.txt' ), 'wb' ) as f :
256- f .write ('1 abandonned 1\n ' . encode ( 'utf-8' ) )
254+ f .write (b '1 abandonned 1\n ' )
257255 assert cs .main (bad_name ) == 2
258256 assert cs .main ('-x' , f .name , bad_name ) == 1
259257
@@ -266,11 +264,11 @@ def test_encoding(tmpdir, capsys):
266264 # with CaptureStdout() as sio:
267265 assert cs .main (f .name ) == 0
268266 with open (f .name , 'wb' ) as f :
269- f .write (u 'naïve\n ' .encode ('utf-8' ))
267+ f .write ('naïve\n ' .encode ())
270268 assert cs .main (f .name ) == 0
271269 assert cs .main ('-e' , f .name ) == 0
272270 with open (f .name , 'ab' ) as f :
273- f .write (u 'naieve\n '. encode ( 'utf-8' ) )
271+ f .write (b 'naieve\n ' )
274272 assert cs .main (f .name ) == 1
275273 # Encoding detection (only try ISO 8859-1 because UTF-8 is the default)
276274 with open (f .name , 'wb' ) as f :
@@ -395,7 +393,7 @@ def test_case_handling(tmpdir, capsys):
395393 # with CaptureStdout() as sio:
396394 assert cs .main (f .name ) == 0
397395 with open (f .name , 'wb' ) as f :
398- f .write ('this has an ACII error' . encode ( 'utf-8' ) )
396+ f .write (b 'this has an ACII error' )
399397 code , stdout , _ = cs .main (f .name , std = True )
400398 assert code == 1
401399 assert 'ASCII' in stdout
0 commit comments