@@ -82,25 +82,17 @@ def close(self):
8282
8383class BufferSizesTests (BaseTests , unittest .TestCase ):
8484 def test_buffer_sizes (self ):
85- # First, run the tests with default and teeny buffer size.
86- for round , bs in (0 , 0 ), (1 , 30 ):
87- t1 = self .writeTmp ('' .join ("Line %s of file 1\n " % (i + 1 ) for i in range (15 )))
88- t2 = self .writeTmp ('' .join ("Line %s of file 2\n " % (i + 1 ) for i in range (10 )))
89- t3 = self .writeTmp ('' .join ("Line %s of file 3\n " % (i + 1 ) for i in range (5 )))
90- t4 = self .writeTmp ('' .join ("Line %s of file 4\n " % (i + 1 ) for i in range (1 )))
91- if bs :
92- with self .assertWarns (DeprecationWarning ):
93- self .buffer_size_test (t1 , t2 , t3 , t4 , bs , round )
94- else :
95- self .buffer_size_test (t1 , t2 , t3 , t4 , bs , round )
96-
97- def buffer_size_test (self , t1 , t2 , t3 , t4 , bs = 0 , round = 0 ):
85+
86+ t1 = self .writeTmp ('' .join ("Line %s of file 1\n " % (i + 1 ) for i in range (15 )))
87+ t2 = self .writeTmp ('' .join ("Line %s of file 2\n " % (i + 1 ) for i in range (10 )))
88+ t3 = self .writeTmp ('' .join ("Line %s of file 3\n " % (i + 1 ) for i in range (5 )))
89+ t4 = self .writeTmp ('' .join ("Line %s of file 4\n " % (i + 1 ) for i in range (1 )))
90+
9891 pat = re .compile (r'LINE (\d+) OF FILE (\d+)' )
9992
100- start = 1 + round * 6
10193 if verbose :
102- print ('%s . Simple iteration (bs=%s)' % ( start + 0 , bs ) )
103- fi = FileInput (files = (t1 , t2 , t3 , t4 ), bufsize = bs )
94+ print ('1 . Simple iteration' )
95+ fi = FileInput (files = (t1 , t2 , t3 , t4 ))
10496 lines = list (fi )
10597 fi .close ()
10698 self .assertEqual (len (lines ), 31 )
@@ -110,8 +102,8 @@ def buffer_size_test(self, t1, t2, t3, t4, bs=0, round=0):
110102 self .assertEqual (fi .filename (), t4 )
111103
112104 if verbose :
113- print ('%s . Status variables (bs=%s)' % ( start + 1 , bs ) )
114- fi = FileInput (files = (t1 , t2 , t3 , t4 ), bufsize = bs )
105+ print ('2 . Status variables' )
106+ fi = FileInput (files = (t1 , t2 , t3 , t4 ))
115107 s = "x"
116108 while s and s != 'Line 6 of file 2\n ' :
117109 s = fi .readline ()
@@ -122,15 +114,15 @@ def buffer_size_test(self, t1, t2, t3, t4, bs=0, round=0):
122114 self .assertFalse (fi .isstdin ())
123115
124116 if verbose :
125- print ('%s . Nextfile (bs=%s)' % ( start + 2 , bs ) )
117+ print ('3 . Nextfile' )
126118 fi .nextfile ()
127119 self .assertEqual (fi .readline (), 'Line 1 of file 3\n ' )
128120 self .assertEqual (fi .lineno (), 22 )
129121 fi .close ()
130122
131123 if verbose :
132- print ('%s . Stdin (bs=%s)' % ( start + 3 , bs ) )
133- fi = FileInput (files = (t1 , t2 , t3 , t4 , '-' ), bufsize = bs )
124+ print ('4 . Stdin' )
125+ fi = FileInput (files = (t1 , t2 , t3 , t4 , '-' ))
134126 savestdin = sys .stdin
135127 try :
136128 sys .stdin = StringIO ("Line 1 of stdin\n Line 2 of stdin\n " )
@@ -143,27 +135,27 @@ def buffer_size_test(self, t1, t2, t3, t4, bs=0, round=0):
143135 sys .stdin = savestdin
144136
145137 if verbose :
146- print ('%s . Boundary conditions (bs=%s)' % ( start + 4 , bs ) )
147- fi = FileInput (files = (t1 , t2 , t3 , t4 ), bufsize = bs )
138+ print ('5 . Boundary conditions' )
139+ fi = FileInput (files = (t1 , t2 , t3 , t4 ))
148140 self .assertEqual (fi .lineno (), 0 )
149141 self .assertEqual (fi .filename (), None )
150142 fi .nextfile ()
151143 self .assertEqual (fi .lineno (), 0 )
152144 self .assertEqual (fi .filename (), None )
153145
154146 if verbose :
155- print ('%s . Inplace (bs=%s)' % ( start + 5 , bs ) )
147+ print ('6 . Inplace' )
156148 savestdout = sys .stdout
157149 try :
158- fi = FileInput (files = (t1 , t2 , t3 , t4 ), inplace = 1 , bufsize = bs )
150+ fi = FileInput (files = (t1 , t2 , t3 , t4 ), inplace = 1 )
159151 for line in fi :
160152 line = line [:- 1 ].upper ()
161153 print (line )
162154 fi .close ()
163155 finally :
164156 sys .stdout = savestdout
165157
166- fi = FileInput (files = (t1 , t2 , t3 , t4 ), bufsize = bs )
158+ fi = FileInput (files = (t1 , t2 , t3 , t4 ))
167159 for line in fi :
168160 self .assertEqual (line [- 1 ], '\n ' )
169161 m = pat .match (line [:- 1 ])
@@ -533,12 +525,11 @@ def test_pathlib_file_inplace(self):
533525class MockFileInput :
534526 """A class that mocks out fileinput.FileInput for use during unit tests"""
535527
536- def __init__ (self , files = None , inplace = False , backup = "" , bufsize = 0 ,
528+ def __init__ (self , files = None , inplace = False , backup = "" , * ,
537529 mode = "r" , openhook = None ):
538530 self .files = files
539531 self .inplace = inplace
540532 self .backup = backup
541- self .bufsize = bufsize
542533 self .mode = mode
543534 self .openhook = openhook
544535 self ._file = None
@@ -641,13 +632,11 @@ def do_test_call_input(self):
641632 files = object ()
642633 inplace = object ()
643634 backup = object ()
644- bufsize = object ()
645635 mode = object ()
646636 openhook = object ()
647637
648638 # call fileinput.input() with different values for each argument
649639 result = fileinput .input (files = files , inplace = inplace , backup = backup ,
650- bufsize = bufsize ,
651640 mode = mode , openhook = openhook )
652641
653642 # ensure fileinput._state was set to the returned object
@@ -658,7 +647,6 @@ def do_test_call_input(self):
658647 self .assertIs (files , result .files , "files" )
659648 self .assertIs (inplace , result .inplace , "inplace" )
660649 self .assertIs (backup , result .backup , "backup" )
661- self .assertIs (bufsize , result .bufsize , "bufsize" )
662650 self .assertIs (mode , result .mode , "mode" )
663651 self .assertIs (openhook , result .openhook , "openhook" )
664652
0 commit comments