added test for readVTKVolume

Shaokun-X · May 4, 2020 · 16c4e5c · 16c4e5c
1 parent cef3cdb
commit 16c4e5c
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/test_vtkutils.py b/tests/test_vtkutils.py
@@ -5,6 +5,7 @@
 import vtk
 import SimpleITK as sitk
 import os
+import create_data
 
 class TestVTKUtils(unittest.TestCase):
 
@@ -81,6 +82,22 @@ def test_meshIO(self):
             self.fail("readMesh failed")
 
 
+    def test_readVTKVolume(self):
+        print("Testing readVTKVolume")
+        tetra = create_data.make_tetra(32)
+        sitk.WriteImage(tetra, "tetra.vtk")
+        try:
+            vtkvol = vtkutils.readVTKVolume("tetra.vtk")
+            print(type(vtkvol))
+            print(vtkvol.GetDimensions())
+        except:
+            sitk.fail("readVTKVolume failed")
+
+        try:
+            os.remove("tetra.vtk")
+        except:
+            print("remove tetra.vtk failed")
+
 if __name__ == "__main__":
     unittest.main()