test

Author: BotellaA

src/opengeodeweb_viewer/rpc/viewer/viewer_protocols.py

@@ -13,6 +13,7 @@
     vtkAbstractMapper,
     vtkWorldPointPicker,
 )
+from vtkmodules.vtkInteractionStyle import vtkInteractorStyleTrackball
 from vtkmodules.vtkCommonCore import reference
 from vtkmodules.vtkCommonDataModel import vtkBoundingBox
 from vtkmodules.vtkCommonTransforms import vtkTransform
@@ -75,7 +76,8 @@ def resetVisualization(self, rpc_params: RpcParams) -> None:
 
         renderWindowInteractor = vtkRenderWindowInteractor()
         renderWindowInteractor.SetRenderWindow(renderWindow)
-        renderWindowInteractor.GetInteractorStyle().SetCurrentStyleToTrackballCamera()
+        style = vtkInteractorStyleTrackball()
+        renderWindowInteractor.SetInteractorStyle(style)
         renderWindowInteractor.EnableRenderOff()
         widget = vtkOrientationMarkerWidget()
         widget.SetInteractor(renderWindowInteractor)

src/opengeodeweb_viewer/vtk_protocol.py

@@ -4,6 +4,7 @@
 from dataclasses import dataclass, field
 
 # Third party imports
+import vtkmodules.vtkRenderingVolumeOpenGL2
 from vtkmodules.web import protocols as vtk_protocols
 from vtkmodules.vtkIOXML import (
     vtkXMLReader,

tests/conftest.py

@@ -3,7 +3,8 @@
 from websocket import create_connection, WebSocketTimeoutException
 import json
 from xprocess import ProcessStarter
-import vtk
+from vtkmodules.vtkIOImage import vtkImageReader2, vtkPNGReader, vtkJPEGReader
+from vtkmodules.vtkImagingCore import vtkImageDifference
 import os
 import shutil
 import xml.etree.ElementTree as ET
@@ -69,28 +70,28 @@ def get_response(self) -> bytes | dict[str, object] | str:
             return str(response)
 
     @staticmethod
-    def _reader_for_file(path: str) -> vtk.vtkImageReader2:
+    def _reader_for_file(path: str) -> vtkImageReader2:
         lower = path.lower()
         if lower.endswith(".png"):
-            return vtk.vtkPNGReader()
+            return vtkPNGReader()
         if lower.endswith(".jpg") or lower.endswith(".jpeg"):
-            return vtk.vtkJPEGReader()
-        return vtk.vtkJPEGReader()
+            return vtkJPEGReader()
+        return vtkJPEGReader()
 
     def images_diff(self, first_image_path: str, second_image_path: str) -> float:
         if ".png" in first_image_path:
-            first_reader = vtk.vtkPNGReader()
+            first_reader = vtkPNGReader()
         elif (".jpg" in first_image_path) or (".jpeg" in first_image_path):
-            first_reader = vtk.vtkJPEGReader()
+            first_reader = vtkJPEGReader()
         first_reader.SetFileName(first_image_path)
 
         if ".png" in second_image_path:
-            second_reader = vtk.vtkPNGReader()
+            second_reader = vtkPNGReader()
         elif (".jpg" in second_image_path) or (".jpeg" in second_image_path):
-            second_reader = vtk.vtkJPEGReader()
+            second_reader = vtkJPEGReader()
         second_reader.SetFileName(second_image_path)
 
-        images_diff = vtk.vtkImageDifference()
+        images_diff = vtkImageDifference()
         images_diff.SetInputConnection(first_reader.GetOutputPort())
         images_diff.SetImageConnection(second_reader.GetOutputPort())
         images_diff.Update()