GH17778: add 'zip' format to unittests.

Author: kchomski

pandas/tests/io/test_pickle.py

@@ -382,7 +382,7 @@ def decompress_file(self, src_path, dest_path, compression):
             fh.write(f.read())
         f.close()
 
-    @pytest.mark.parametrize('compression', [None, 'gzip', 'bz2', 'xz'])
+    @pytest.mark.parametrize('compression', [None, 'gzip', 'zip', 'bz2', 'xz'])
     def test_write_explicit(self, compression, get_random_path):
         # issue 11666
         if compression == 'xz':
@@ -414,7 +414,8 @@ def test_write_explicit_bad(self, compression, get_random_path):
                 df = tm.makeDataFrame()
                 df.to_pickle(path, compression=compression)
 
-    @pytest.mark.parametrize('ext', ['', '.gz', '.bz2', '.xz', '.no_compress'])
+    @pytest.mark.parametrize('ext', ['', '.gz', '.zip', '.bz2', '.xz',
+                                     '.no_compress'])
     def test_write_infer(self, ext, get_random_path):
         if ext == '.xz':
             tm._skip_if_no_lzma()
@@ -442,7 +443,7 @@ def test_write_infer(self, ext, get_random_path):
 
             tm.assert_frame_equal(df, df2)
 
-    @pytest.mark.parametrize('compression', [None, 'gzip', 'bz2', 'xz', "zip"])
+    @pytest.mark.parametrize('compression', [None, 'gzip', 'bz2', 'xz', 'zip'])
     def test_read_explicit(self, compression, get_random_path):
         # issue 11666
         if compression == 'xz':