Merge pull request #36 from MuellerSeb/fix_low_dim_vti

imageToVTK: determine low-dim grid from spacing

Author: renefritze

.github/workflows/test.yml

@@ -13,7 +13,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.6", "3.7", "3.8", "3.9", "3.10"]
+        python-version: ["3.7", "3.8", "3.9", "3.10", "3.11"]
 
     env:
       SDIST_DIR: /tmp/sdist

pyevtk/hl.py

@@ -202,6 +202,12 @@ def imageToVTK(
             end = data.shape
         elif data[0].ndim == 3 and data[1].ndim == 3 and data[2].ndim == 3:
             end = data[0].shape
+        for i, s in enumerate(spacing):
+            if np.isclose(s, 0.0):
+                if end[i] == 1:
+                    end = end[:i] + (0,) + end[i + 1 :]
+                else:
+                    raise ValueError("imageToVTK: grid has lower dimension than data")
     elif pointData is not None:
         keys = list(pointData.keys())
         data = pointData[keys[0]]
@@ -210,6 +216,9 @@ def imageToVTK(
         elif data[0].ndim == 3 and data[1].ndim == 3 and data[2].ndim == 3:
             end = data[0].shape
         end = (end[0] - 1, end[1] - 1, end[2] - 1)
+        for i, s in enumerate(spacing):
+            if np.isclose(s, 0.0) and end[i] > 0:
+                raise ValueError("imageToVTK: grid has lower dimension than data")
 
     # Write data to file
     w = VtkFile(path, VtkImageData)

pyevtk/vtk.py

@@ -36,6 +36,7 @@
 #            VTK Types
 # ================================
 
+
 #     FILE TYPES
 class VtkFileType:
     """

setup.py

@@ -57,9 +57,10 @@ def readme(fname):
     setup_requires=["pytest-runner"],
     tests_require=["pytest>=3.1", "pytest-cov", "twine", "check-manifest"],
     classifiers=[
-        "Programming Language :: Python :: 3.6",
         "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
     ],
 )