Added npy format support to diffread

CHANGELOG.rst

@@ -5,6 +5,7 @@ Release 2.1.2
 -------------
 
 * Improved :func:`autocenter` for diffraction patterns with large Ewald sphere walkoff.
+* :func:`diffread` now supports NumPy's ``*.npy`` format.
 * Speedup of all routines that use the Fast Fourier transform (:func:`autocenter`, :func:`align`, :func:`ialign`, :func:`itrack_peak`, and :func:`kinematicsim`) by 50%.
 
 Release 2.1.1

skued/io/io.py

@@ -2,6 +2,8 @@
 
 from pathlib import Path
 
+import numpy as np
+
 from .dm import dmread
 from .merlin import mibread
 
@@ -18,6 +20,8 @@ def diffread(fname):
 
         * TIFF (`*.tif`, `*.tiff`)
 
+        * NumPy (`*.npy`)
+
         * All file formats supported by Scikit-image
 
     Parameters
@@ -42,6 +46,8 @@ def diffread(fname):
         return mibread(fname)
     elif fname.suffix in {".dm3", ".dm4"}:
         return dmread(fname)
+    elif fname.suffix == ".npy":
+        return np.load(fname, fix_imports=False)
 
     # Worst-case scenario, we need skimage.io
     # since this is a slow import, we import it here

skued/io/tests/test_io.py

@@ -6,6 +6,7 @@
 
 import numpy as np
 from skimage.io import imsave
+import tempfile
 
 from skued import diffread, dmread, imibread, mibheader, mibread
 from skued.utils import suppress_warnings
@@ -54,6 +55,15 @@ def test_diffread_on_tiff():
     os.remove(path)
 
 
+def test_diffread_on_npy():
+    """ Test diffread() on *.npy files """
+    arr = np.random.random(size=(128, 128))
+    with tempfile.TemporaryDirectory() as tmpdir:
+        np.save(Path(tmpdir) / "skued_test.npy", arr)
+        arr2 = diffread(Path(tmpdir) / "skued_test.npy")
+    assert np.allclose(arr, arr2)
+
+
 def test_diffread_on_skimage_png():
     """ Test the last resort of using skimage.io for pngs """
     from_skimage = diffread(TEST_PNG)