Skip to content

Commit 38f7d54

Browse files
authored
fix pickle test
1 parent 57eba0a commit 38f7d54

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

pandas/tests/io/test_pickle.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -289,9 +289,7 @@ def get_random_path():
289289

290290
class TestCompression:
291291

292-
_extension_to_compression = {
293-
ext: compression for compression, ext in icom._compression_to_extension.items()
294-
}
292+
_extension_to_compression = icom._extension_to_compression
295293

296294
def compress_file(self, src_path, dest_path, compression):
297295
if compression is None:
@@ -550,7 +548,7 @@ def test_pickle_binary_object_compression(compression):
550548
buffer.seek(0)
551549

552550
# gzip and zip safe the filename: cannot compare the compressed content
553-
assert buffer.getvalue() == reference or compression in ("gzip", "zip")
551+
assert buffer.getvalue() == reference or compression in ("gzip", "zip", "tar")
554552

555553
# read
556554
read_df = pd.read_pickle(buffer, compression=compression)

0 commit comments

Comments
 (0)