|
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