diff options
Diffstat (limited to 'numpy/compat/tests/test_compat.py')
-rw-r--r-- | numpy/compat/tests/test_compat.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/numpy/compat/tests/test_compat.py b/numpy/compat/tests/test_compat.py new file mode 100644 index 000000000..9822ab374 --- /dev/null +++ b/numpy/compat/tests/test_compat.py @@ -0,0 +1,23 @@ +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() |