Merge topic 'matplotlib-python-test'

051c9661 Add python test to matplotlib extension.

Acked-by: Kitware Robot <kwrobot@kitware.com>
Acked-by: Haocheng Liu <haocheng.liu@kitware.com>
Merge-request: !719

smtk/extension/matplotlib/__init__.py

@@ -9,3 +9,5 @@
 #  PURPOSE.  See the above copyright notice for more information.
 #
 #=============================================================================
+
+import render_mesh

smtk/extension/matplotlib/testing/CMakeLists.txt

 add_subdirectory(cxx)
+add_subdirectory(python)

smtk/extension/matplotlib/testing/cxx/RenderMesh.cxx

@@ -106,7 +106,7 @@ int main(int argc, char* argv[])
   // Set the mesh to be rendered
   op->specification()->findMesh("mesh")->setValue(c->meshes());
 
-  // Set the file path for the rendered imagel
+  // Set the file path for the rendered image
   std::string write_path(write_root);
   write_path += "/" + smtk::common::UUID::random().toString() + ".png";
   op->specification()->findFile("filename")->setValue(write_path);

smtk/extension/matplotlib/testing/python/CMakeLists.txt

+set(smtkMatplotlibExtPythonDataTests)
+
+if (SMTK_USE_PYBIND11)
+  set(smtkMatplotlibExtPythonDataTests
+    ${smtkMatplotlibExtPythonDataTests}
+    renderMesh
+  )
+endif()
+
+if (SMTK_DATA_DIR AND EXISTS ${SMTK_DATA_DIR}/cmb-testing-data.marker)
+  foreach (test ${smtkMatplotlibExtPythonDataTests})
+    smtk_add_test_python(${test}Py ${test}.py
+      --data-dir=${SMTK_DATA_DIR})
+    set_tests_properties(${test}Py PROPERTIES LABELS "ExtMatplotlib")
+  endforeach()
+endif()

smtk/extension/matplotlib/testing/python/renderMesh.py

+#=============================================================================
+#
+#  Copyright (c) Kitware, Inc.
+#  All rights reserved.
+#  See LICENSE.txt for details.
+#
+#  This software is distributed WITHOUT ANY WARRANTY; without even
+#  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
+#  PURPOSE.  See the above copyright notice for more information.
+#
+#=============================================================================
+import os
+import sys
+import unittest
+import smtk
+if smtk.wrappingProtocol() == 'pybind11':
+    import smtk.mesh
+    import smtk.model
+    import smtk.extension.matplotlib
+import smtk.testing
+from smtk.simple import *
+
+
+class RenderMesh(smtk.testing.TestCase):
+
+    def setUp(self):
+        self.mgr = smtk.model.Manager.create()
+        self.meshmgr = self.mgr.meshes()
+        self.sess = self.mgr.createSession('native')
+        SetActiveSession(self.sess)
+        self.meshFile = os.path.join(
+            smtk.testing.DATA_DIR, 'mesh', '2d', 'warpedMesh.h5m')
+        self.collection = smtk.io.importMesh(self.meshFile, self.meshmgr)
+
+    def testRenderMesh(self):
+        renderMesh = self.sess.op('render mesh')
+        renderMesh.specification().findMesh(
+            'mesh').setValue(self.collection.meshes())
+        self.outFile = os.path.join(
+            smtk.testing.TEMP_DIR, str(smtk.common.UUID.random()) + '.png')
+        renderMesh.specification().findFile('filename').setValue(self.outFile)
+        result = renderMesh.operate()
+
+        if self.interactive() and self.haveVTK() and self.haveVTKExtension():
+            import vtk
+            self.startRenderTest()
+            reader = vtk.vtkPNGReader()
+            reader.SetFileName(self.outFile)
+            reader.Update()
+            self.addImageToScene(reader)
+            cam = self.renderer.GetActiveCamera()
+            self.renderer.ResetCamera()
+            self.renderWindow.Render()
+            self.interact()
+
+
+if __name__ == '__main__':
+    smtk.testing.process_arguments()
+    unittest.main()

smtk/smtk/testing.py

@@ -112,6 +112,7 @@ def compare_image(render_window, baseline_path):
 
 
 class TestCaseMeta(type):
+
     """A metaclass for tests.
 
     This is used to make TestCase inherit vtk.test.Testing.vtkTest
@@ -214,6 +215,14 @@ class TestCase:
         else:
             print "Shiboken does not support vtk/smtk compatibility layer"
 
+    def addImageToScene(self, msource):
+        import vtk
+        vsource = msource
+        self.renderWindow = vtk.vtkImageViewer2()
+        self.renderer = self.renderWindow.GetRenderer()
+        self.renderWindow.SetInputConnection(vsource.GetOutputPort())
+        self.renderWindow.SetupInteractor(self.interactor)
+
     def interactive(self):
         """Return false if the test should exit at completion."""
         global INTERACTIVE