Skip to content

Commit d0d21b4

Browse files
author
Pankaj Pandey
committed
Use tm.capture_stderr to capture stderr
1 parent 5d112a3 commit d0d21b4

File tree

1 file changed

+7
-10
lines changed

1 file changed

+7
-10
lines changed

pandas/tests/io/parser/comment.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@ def test_commment_first_line(self):
119119
result = self.read_csv(StringIO(data), comment='#', header=None)
120120
tm.assert_frame_equal(result, expected)
121121

122+
@tm.capture_stderr
122123
def test_comment_whitespace_delimited(self):
123124
test_input = """\
124125
1 2
@@ -131,16 +132,12 @@ def test_comment_whitespace_delimited(self):
131132
8# 1 field, NaN
132133
9 2 3 # skipped line
133134
# 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()
143139
# 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
145142
expected = DataFrame([[1, 2], [5, 2], [6, 2], [7, np.nan], [8, np.nan]])
146143
tm.assert_frame_equal(df, expected)

0 commit comments

Comments
 (0)