Skip to content
Prev Previous commit
Next Next commit
add compression size test case
  • Loading branch information
minggli committed May 25, 2018
commit 012383f7c443ee710e7bccf9d5a109f7b86743c4
18 changes: 18 additions & 0 deletions pandas/tests/io/json/test_compression.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,24 @@ def test_compression_roundtrip(compression):
assert_frame_equal(df, pd.read_json(result))


def test_to_csv_compression_size(compression):

df = pd.concat(100 * [pd.DataFrame([[0.123456, 0.234567, 0.567567],
[12.32112, 123123.2, 321321.2]],
columns=['X', 'Y', 'Z'])],
ignore_index=True)

with tm.ensure_clean() as filename:
import os
df.to_json(filename, compression=compression)
file_size = os.path.getsize(filename)

if compression:
df.to_json(filename, compression=None)
uncompressed_file_size = os.path.getsize(filename)
assert uncompressed_file_size > file_size


def test_read_zipped_json():
uncompressed_path = tm.get_data_path("tsframe_v012.json")
uncompressed_df = pd.read_json(uncompressed_path)
Expand Down