ENH: Throw in .zst while here

Author: effigies

nibabel/filebasedimages.py

@@ -430,7 +430,7 @@ def _sniff_meta_for(klass, filename, sniff_nbytes, sniff=None):
         try:
             with ImageOpener(meta_fname, 'rb') as fobj:
                 binaryblock = fobj.read(sniff_nbytes)
-        except IOError:
+        except (IOError, EOFError):
             return None
         return (binaryblock, meta_fname)
 

nibabel/filename_parser.py

@@ -253,7 +253,7 @@ def _iendswith(whole, end):
 
 
 def splitext_addext(filename,
-                    addexts=('.gz', '.bz2'),
+                    addexts=('.gz', '.bz2', '.zst'),
                     match_case=False):
     """ Split ``/pth/fname.ext.gz`` into ``/pth/fname, .ext, .gz``
 

nibabel/loadsave.py

@@ -44,7 +44,8 @@ def _signature_matches_extension(filename):
     """
     signatures = {
         ".gz": {"signature": b"\x1f\x8b", "format_name": "gzip"},
-        ".bz2": {"signature": b"BZh", "format_name": "bzip2"}
+        ".bz2": {"signature": b"BZh", "format_name": "bzip2"},
+        ".zst": {"signature": b"\x28\xb5\x2f\xfd", "format_name": "ztsd"},
     }
     filename = _stringify_path(filename)
     *_, ext = splitext_addext(filename)

nibabel/tests/test_loadsave.py

@@ -17,6 +17,7 @@
 
 from ..optpkg import optional_package
 _, have_scipy, _ = optional_package('scipy')
+_, have_pyzstd, _ = optional_package('pyzstd')
 
 from numpy.testing import (assert_almost_equal,
                            assert_array_equal)
@@ -75,16 +76,20 @@ def test_load_empty_image():
     assert str(err.value).startswith('Empty file: ')
 
 
-@pytest.mark.parametrize("extension", [".gz", ".bz2"])
+@pytest.mark.parametrize("extension", [".gz", ".bz2", ".zst"])
 def test_load_bad_compressed_extension(tmp_path, extension):
+    if extension == ".zst" and not have_pyzstd:
+        pytest.skip()
     file_path = tmp_path / f"img.nii{extension}"
     file_path.write_bytes(b"bad")
     with pytest.raises(ImageFileError, match=".*is not a .* file"):
         load(file_path)
 
 
-@pytest.mark.parametrize("extension", [".gz", ".bz2"])
+@pytest.mark.parametrize("extension", [".gz", ".bz2", ".zst"])
 def test_load_good_extension_with_bad_data(tmp_path, extension):
+    if extension == ".zst" and not have_pyzstd:
+        pytest.skip()
     file_path = tmp_path / f"img.nii{extension}"
     with Opener(file_path, "wb") as fobj:
         fobj.write(b"bad")