Program Listing for File main.py

↰ Return to documentation for file (tests/packages/build_module/main.py)

# This tests that we can use our custom vtkMRMLLayerDMPipelineI subclass
# with the vtkMRMLLayerDMPipelineScriptedCreator and vtkMRMLLayerDMPipelineFactory

from slicer import (
    vtkMRMLLayerDMPipelineFactory,
    vtkMRMLDummyLayerDMPipeline,
    vtkMRMLLayerDMPipelineScriptedCreator
)


factory = vtkMRMLLayerDMPipelineFactory()
creator = vtkMRMLLayerDMPipelineScriptedCreator()
creator.SetPythonCallback(lambda *_ : vtkMRMLDummyLayerDMPipeline())
factory.AddPipelineCreator(creator)
assert(isinstance(factory.CreatePipeline(None, None), vtkMRMLDummyLayerDMPipeline))