Support squashing files with Z guards

Author: ZedThree

src/boutdata/data.py

@@ -1044,6 +1044,7 @@ def __init__(
         info=False,
         xguards=True,
         yguards=False,
+        zguards=False,
         tind=None,
         xind=None,
         yind=None,
@@ -1063,6 +1064,7 @@ def __init__(
         self._info = info
         self._xguards = xguards
         self._yguards = yguards
+        self._zguards = zguards
         self._kwargs = kwargs
         self._parallel = parallel
         if suffix is None:
@@ -1111,7 +1113,7 @@ def __init__(
                 f,
                 xguards=self._xguards,
                 yguards=self._yguards,
-                zguards=False,
+                zguards=self._zguards,
                 tind=tind,
                 xind=xind,
                 yind=yind,
@@ -1523,6 +1525,7 @@ def _collect(self, varname):
             info=self._info,
             xguards=self._xguards,
             yguards=self._yguards,
+            zguards=self._zguards,
             tind=self.tind,
             xind=self.xind,
             yind=self.yind,

src/boutdata/squashoutput.py

@@ -23,6 +23,7 @@ def squashoutput(
     zind=None,
     xguards=True,
     yguards="include_upper",
+    zguards=True,
     drop_variables=None,
     singleprecision=False,
     compress=False,
@@ -156,6 +157,7 @@ def squashoutput(
         info=False,
         xguards=xguards,
         yguards=yguards,
+        zguards=zguards,
         tind=tind,
         xind=xind,
         yind=yind,