Skip to content

Commit 5df1618

Browse files
committed
fix Series.to_csv compression tests
1 parent d9ac263 commit 5df1618

File tree

1 file changed

+7
-14
lines changed

1 file changed

+7
-14
lines changed

pandas/tests/series/test_io.py

+7-14
Original file line numberDiff line numberDiff line change
@@ -139,12 +139,6 @@ def test_to_csv_path_is_none(self):
139139
csv_str = s.to_csv(path=None)
140140
assert isinstance(csv_str, str)
141141

142-
@pytest.mark.parametrize('compression', [
143-
None,
144-
'gzip',
145-
'bz2',
146-
pytest.param('xz', marks=td.skip_if_no_lzma),
147-
])
148142
def test_to_csv_compression(self, compression):
149143

150144
s = Series([0.123456, 0.234567, 0.567567], index=['A', 'B', 'C'],
@@ -160,14 +154,13 @@ def test_to_csv_compression(self, compression):
160154
assert_series_equal(s, rs)
161155

162156
# explicitly ensure file was compressed
163-
f = tm.decompress_file(filename, compression=compression)
164-
text = f.read().decode('utf8')
165-
assert s.name in text
166-
f.close()
167-
168-
f = tm.decompress_file(filename, compression=compression)
169-
assert_series_equal(s, pd.read_csv(f, index_col=0, squeeze=True))
170-
f.close()
157+
with tm.decompress_file(filename, compression=compression) as fh:
158+
text = fh.read().decode('utf8')
159+
assert s.name in text
160+
161+
with tm.decompress_file(filename, compression=compression) as fh:
162+
assert_series_equal(s, pd.read_csv(fh,
163+
index_col=0, squeeze=True))
171164

172165

173166
class TestSeriesIO(TestData):

0 commit comments

Comments
 (0)