TEST: Updated to expect indexed_gzip if present

Author: pauldmccarthy

nibabel/tests/test_openers.py

@@ -12,9 +12,10 @@
 from gzip import GzipFile
 from bz2 import BZ2File
 from io import BytesIO, UnsupportedOperation
+from distutils.version import StrictVersion
 
 from ..py3k import asstr, asbytes
-from ..openers import Opener, ImageOpener
+from ..openers import Opener, ImageOpener, HAVE_INDEXED_GZIP
 from ..tmpdirs import InTemporaryDirectory
 from ..volumeutils import BinOpener
 
@@ -67,6 +68,8 @@ def test_Opener_various():
     # Check we can do all sorts of files here
     message = b"Oh what a giveaway"
     bz2_fileno = hasattr(BZ2File, 'fileno')
+    if HAVE_INDEXED_GZIP:
+        import indexed_gzip as igzip
     with InTemporaryDirectory():
         sobj = BytesIO()
         for input in ('test.txt',
@@ -86,6 +89,11 @@ def test_Opener_various():
                     assert_raises(UnsupportedOperation, fobj.fileno)
                 elif input.endswith('.bz2') and not bz2_fileno:
                     assert_raises(AttributeError, fobj.fileno)
+                # indexed gzip is used by default, and drops file
+                # handles by default, so we don't have a fileno.
+                elif input.endswith('gz') and HAVE_INDEXED_GZIP and \
+                     StrictVersion(igzip.__version__) >= StrictVersion('0.7.0'):
+                    assert_raises(igzip.NoHandleError, fobj.fileno)
                 else:
                     # Just check there is a fileno
                     assert_not_equal(fobj.fileno(), 0)
@@ -139,7 +147,7 @@ def test_Opener_gzip_type():
             (False, {'mode' : 'wb', 'keep_open' : False},  GzipFile),
             (False, {'mode' : 'wb', 'keep_open' : 'auto'}, GzipFile),
             (True,  {'mode' : 'rb', 'keep_open' : True},   MockIndexedGzipFile),
-            (True,  {'mode' : 'rb', 'keep_open' : False},  GzipFile),
+            (True,  {'mode' : 'rb', 'keep_open' : False},  MockIndexedGzipFile),
             (True,  {'mode' : 'rb', 'keep_open' : 'auto'}, MockIndexedGzipFile),
             (True,  {'mode' : 'wb', 'keep_open' : True},   GzipFile),
             (True,  {'mode' : 'wb', 'keep_open' : False},  GzipFile),
@@ -260,11 +268,10 @@ class StrictOpener(Opener):
                 assert_true(isinstance(fobj.fobj, file_class))
             elif lext == 'gz':
                 try:
-                    from indexed_gzip import SafeIndexedGzipFile
+                    from ..openers import IndexedGzipFile
                 except ImportError:
-                    SafeIndexedGzipFile = GzipFile
-                assert_true(isinstance(fobj.fobj, (GzipFile,
-                                                   SafeIndexedGzipFile)))
+                    IndexedGzipFile = GzipFile
+                assert_true(isinstance(fobj.fobj, (GzipFile, IndexedGzipFile)))
             else:
                 assert_true(isinstance(fobj.fobj, BZ2File))