summaryrefslogtreecommitdiff
path: root/numpy/compat/tests/test_compat.py
blob: 1ac24401a7192c54981c51e8a97910c964f9cfe0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from __future__ import division, absolute_import, print_function

from os.path import join

from numpy.compat import isfileobj
from numpy.testing import assert_, run_module_suite
from numpy.testing.utils import tempdir


def test_isfileobj():
    with tempdir(prefix="numpy_test_compat_") as folder:
        filename = join(folder, 'a.bin')

        with open(filename, 'wb') as f:
            assert_(isfileobj(f))

        with open(filename, 'ab') as f:
            assert_(isfileobj(f))

        with open(filename, 'rb') as f:
            assert_(isfileobj(f))


if __name__ == "__main__":
    run_module_suite()