tests xfail on csv zip compression in Python 2

Author: minggli

pandas/tests/frame/test_to_csv.py

@@ -8,7 +8,7 @@
 from numpy import nan
 import numpy as np
 
-from pandas.compat import (lmap, range, lrange, StringIO, u)
+from pandas.compat import (lmap, range, lrange, StringIO, u, PY2)
 import pandas.core.common as com
 from pandas.errors import ParserError
 from pandas import (DataFrame, Index, Series, MultiIndex, Timestamp,
@@ -925,6 +925,10 @@ def test_to_csv_compression(self, compression):
                         [12.32112, 123123.2, 321321.2]],
                        index=['A', 'B'], columns=['X', 'Y', 'Z'])
 
+        if PY2 and compression == 'zip':
+            pytest.xfail(reason='zip compression for csv not suppported in'
+                                'Python 2')
+
         with ensure_clean() as filename:
 
             df.to_csv(filename, compression=compression)

pandas/tests/series/test_io.py

@@ -10,7 +10,7 @@
 
 from pandas import Series, DataFrame
 
-from pandas.compat import StringIO, u
+from pandas.compat import StringIO, u, PY2
 from pandas.util.testing import (assert_series_equal, assert_almost_equal,
                                  assert_frame_equal, ensure_clean)
 import pandas.util.testing as tm
@@ -143,6 +143,10 @@ def test_to_csv_compression(self, compression):
         s = Series([0.123456, 0.234567, 0.567567], index=['A', 'B', 'C'],
                    name='X')
 
+        if PY2 and compression == 'zip':
+            pytest.xfail(reason='zip compression for csv not suppported in'
+                                'Python 2')
+
         with ensure_clean() as filename:
 
             s.to_csv(filename, compression=compression, header=True)