Add zst fileslice benchmark

Author: vanandrew

nibabel/benchmarks/bench_fileslice.py

@@ -19,6 +19,12 @@
 from ..rstutils import rst_table
 from ..tmpdirs import InTemporaryDirectory
 
+try:
+    import pyzstd
+    ZSTD_INSTALLED = True
+except ImportError:
+    ZSTD_INSTALLED = False
+
 SHAPE = (64, 64, 32, 100)
 ROW_NAMES = [f'axis {i}, len {dim}' for i, dim in enumerate(SHAPE)]
 COL_NAMES = ['mid int',
@@ -70,7 +76,8 @@ def g():
 def bench_fileslice(bytes=True,
                     file_=True,
                     gz=True,
-                    bz2=False):
+                    bz2=False,
+                    zst=True):
     sys.stdout.flush()
     repeat = 2
 
@@ -103,4 +110,11 @@ def my_table(title, times, base):
         my_table('bz2 slice - raw (ratio)',
                  np.dstack((bz2_times, bz2_times / bz2_base)),
                  bz2_base)
+    if ZSTD_INSTALLED:
+        if zst:
+            with InTemporaryDirectory():
+                zst_times, zst_base = run_slices('data.zst', repeat)
+            my_table('zst slice - raw (ratio)',
+                    np.dstack((zst_times, zst_times / zst_base)),
+                    zst_base)
     sys.stdout.flush()