BUG: fixed errors in python2

pandas-dev · Jun 24, 2018 · 48f5685 · 48f5685
1 parent 050e8b5
commit 48f5685
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pandas/io/common.py b/pandas/io/common.py
@@ -4,7 +4,6 @@
 import csv
 import codecs
 import mmap
-import io
 from contextlib import contextmanager, closing
 import zipfile
 
@@ -395,7 +394,9 @@ def _get_handle(path_or_buf, mode, encoding=None, compression=None,
     elif is_path:
         if compat.PY2:
             # Python 2
-            f = io.open(path_or_buf, mode, newline='\n')
+            if mode == 'w':
+                mode = 'wb'
+            f = open(path_or_buf, mode)
         elif encoding:
             # Python 3 and encoding
             f = open(path_or_buf, mode, encoding=encoding, newline='\n')

diff --git a/pandas/tests/io/formats/test_to_csv.py b/pandas/tests/io/formats/test_to_csv.py
@@ -307,7 +307,7 @@ def test_to_csv_string_with_lf(self):
                 assert f.read() == expected_bin
 
             # 'line_terminator' should not change inner element
-            expected_bin =(
+            expected_bin = (
                 b'int,str_lf\r\n'
                 b'1,abc\r\n'
                 b'2,"d\nef"\r\n'