File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -289,9 +289,7 @@ def get_random_path():
289
289
290
290
class TestCompression :
291
291
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
295
293
296
294
def compress_file (self , src_path , dest_path , compression ):
297
295
if compression is None :
@@ -550,7 +548,7 @@ def test_pickle_binary_object_compression(compression):
550
548
buffer .seek (0 )
551
549
552
550
# 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" )
554
552
555
553
# read
556
554
read_df = pd .read_pickle (buffer , compression = compression )
You can’t perform that action at this time.
0 commit comments