Skip to content

Commit b9c8a24

Browse files
kinowBruno P. Kinoshita
authored and
Bruno P. Kinoshita
committed
Use StringIO instead of capturing sys out
1 parent 8db85ca commit b9c8a24

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

pandas/tests/io/formats/test_to_csv.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -572,10 +572,10 @@ def test_to_csv_na_rep_long_string(self, capsys):
572572
'mynull',
573573
'mynull',
574574
'mynull']
575-
expected_ascii = tm.convert_rows_list_to_csv_str(expected_rows)
575+
expected = tm.convert_rows_list_to_csv_str(expected_rows)
576576

577-
df.to_csv(sys.stdout, index=False, na_rep='mynull', encoding='ascii')
578-
captured = capsys.readouterr()
577+
csv_data = StringIO()
578+
df.to_csv(csv_data, index=False, na_rep='mynull', encoding='ascii')
579+
result = csv_data.getvalue()
579580

580-
assert captured.out == expected_ascii
581-
assert not sys.stdout.closed
581+
assert expected == result

0 commit comments

Comments
 (0)