summaryrefslogtreecommitdiff
path: root/numpy/compat/py3k.py
blob: e99b29deb11b63ff580e865f3b5f81121e55ba1a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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)