Skip to content

Commit b8c4175

Browse files
committed
add compressed pickle data file to io/tests
1 parent 6df6611 commit b8c4175

File tree

5 files changed

+14
-0
lines changed

5 files changed

+14
-0
lines changed
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.

pandas/io/tests/test_pickle.py

+14
Original file line numberDiff line numberDiff line change
@@ -324,6 +324,20 @@ def test_compression_infer(self):
324324
for ext in extensions:
325325
yield self.compression_infer, ext
326326

327+
def compression_prepared_data(self, ext):
328+
pickle_path = os.path.join(tm.get_data_path(),
329+
'pickle_compression',
330+
'data.pickle')
331+
compressed_path = pickle_path + ext
332+
data1 = pd.read_pickle(pickle_path)
333+
data2 = pd.read_pickle(compressed_path)
334+
tm.assert_frame_equal(data1, data2)
335+
336+
def test_compression_prepared_data(self):
337+
extensions = ['.gz', '.bz2', '.xz']
338+
for ext in extensions:
339+
yield self.compression_prepared_data, ext
340+
327341

328342
if __name__ == '__main__':
329343
nose.runmodule(argv=[__file__, '-vvs', '-x', '--pdb', '--pdb-failure'],

0 commit comments

Comments
 (0)