@@ -119,6 +119,7 @@ def test_commment_first_line(self):
119
119
result = self .read_csv (StringIO (data ), comment = '#' , header = None )
120
120
tm .assert_frame_equal (result , expected )
121
121
122
+ @tm .capture_stderr
122
123
def test_comment_whitespace_delimited (self ):
123
124
test_input = """\
124
125
1 2
@@ -131,16 +132,12 @@ def test_comment_whitespace_delimited(self):
131
132
8# 1 field, NaN
132
133
9 2 3 # skipped line
133
134
# comment"""
134
- captured_err = StringIO ()
135
- orig_stderr , sys .stderr = sys .stderr , captured_err
136
- try :
137
- df = self .read_csv (StringIO (test_input ), comment = '#' , header = None ,
138
- delimiter = '\\ s+' , skiprows = 0 ,
139
- error_bad_lines = False )
140
- finally :
141
- sys .stderr = orig_stderr
142
- content = captured_err .getvalue ()
135
+ df = self .read_csv (StringIO (test_input ), comment = '#' , header = None ,
136
+ delimiter = '\\ s+' , skiprows = 0 ,
137
+ error_bad_lines = False )
138
+ error = sys .stderr .getvalue ()
143
139
# skipped lines 2, 3, 4, 9
144
- assert content .count ('Skipping line' ) == 4 , content
140
+ for line_num in (2 , 3 , 4 , 9 ):
141
+ assert 'Skipping line {}' .format (line_num ) in error , error
145
142
expected = DataFrame ([[1 , 2 ], [5 , 2 ], [6 , 2 ], [7 , np .nan ], [8 , np .nan ]])
146
143
tm .assert_frame_equal (df , expected )
0 commit comments