|
7 | 7 |
|
8 | 8 | import numpy as np |
9 | 9 |
|
10 | | -import pandas as pd |
11 | 10 | from pandas import Series, DataFrame, Timestamp |
12 | 11 | from pandas.compat import range, lmap |
13 | 12 | import pandas.core.common as com |
@@ -226,20 +225,19 @@ def test_standardize_mapping(): |
226 | 225 | assert isinstance(com.standardize_mapping(dd), partial) |
227 | 226 |
|
228 | 227 |
|
229 | | -@pytest.mark.parametrize('frame', [ |
230 | | - pd.concat(100 * [DataFrame([[0.123456, 0.234567, 0.567567], |
231 | | - [12.32112, 123123.2, 321321.2]], |
232 | | - columns=['X', 'Y', 'Z'])], ignore_index=True), |
233 | | - pd.concat(100 * [Series([0.123456, 0.234567, 0.567567], name='X')], |
234 | | - ignore_index=True)]) |
| 228 | +@pytest.mark.parametrize('obj', [ |
| 229 | + DataFrame(100 * [[0.123456, 0.234567, 0.567567], |
| 230 | + [12.32112, 123123.2, 321321.2]], |
| 231 | + columns=['X', 'Y', 'Z']), |
| 232 | + Series(100 * [0.123456, 0.234567, 0.567567], name='X')]) |
235 | 233 | @pytest.mark.parametrize('method', ['to_pickle', 'to_json', 'to_csv']) |
236 | | -def test_compression_size(frame, method, compression): |
| 234 | +def test_compression_size(obj, method, compression): |
237 | 235 | if not compression: |
238 | 236 | pytest.skip("only test compression case.") |
239 | 237 |
|
240 | 238 | with tm.ensure_clean() as filename: |
241 | | - getattr(frame, method)(filename, compression=compression) |
| 239 | + getattr(obj, method)(filename, compression=compression) |
242 | 240 | compressed = os.path.getsize(filename) |
243 | | - getattr(frame, method)(filename, compression=None) |
| 241 | + getattr(obj, method)(filename, compression=None) |
244 | 242 | uncompressed = os.path.getsize(filename) |
245 | 243 | assert uncompressed > compressed |
0 commit comments