@@ -139,12 +139,6 @@ def test_to_csv_path_is_none(self):
139
139
csv_str = s .to_csv (path = None )
140
140
assert isinstance (csv_str , str )
141
141
142
- @pytest .mark .parametrize ('compression' , [
143
- None ,
144
- 'gzip' ,
145
- 'bz2' ,
146
- pytest .param ('xz' , marks = td .skip_if_no_lzma ),
147
- ])
148
142
def test_to_csv_compression (self , compression ):
149
143
150
144
s = Series ([0.123456 , 0.234567 , 0.567567 ], index = ['A' , 'B' , 'C' ],
@@ -160,14 +154,13 @@ def test_to_csv_compression(self, compression):
160
154
assert_series_equal (s , rs )
161
155
162
156
# 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 ))
171
164
172
165
173
166
class TestSeriesIO (TestData ):
0 commit comments