3K: add some compatibility tools to numpy.compat

1 files changed, 22 insertions, 0 deletions

diff --git a/numpy/compat/py3k.py b/numpy/compat/py3k.py
new file mode 100644
index 000000000..e99b29deb
--- /dev/null
+++ b/numpy/compat/py3k.py
@@ -0,0 +1,22 @@
+"""
+Python 3 compatibility tools.
+
+"""
+
+__all__ = ['bytes', 'asbytes', 'isfile']
+
+import sys
+if sys.version_info[0] >= 3:
+    import io
+    bytes = bytes
+    def asbytes(s):
+        if isinstance(s, bytes):
+            return s
+        return s.encode('iso-8859-1')
+    def isfile(f):
+        return isinstance(f, io.IOBase)
+else:
+    bytes = str
+    asbytes = str
+    def isfile(f):
+        return isinstance(f, file)