def test_vtk_sync_helpers(document, comm):
renWin1 = make_render_window()
renWin2 = make_render_window()
# Create 2 panes to compare each other
pane1 = VTK(renWin1)
pane2 = VTK(renWin2)
assert isinstance(pane1, VTKRenderWindowSynchronized)
assert isinstance(pane2, VTKRenderWindowSynchronized)
# Create get models
model1 = pane1.get_root(document, comm=comm)
model2 = pane2.get_root(document, comm=comm)
assert isinstance(model1, VTKSynchronizedPlot)
assert isinstance(model2, VTKSynchronizedPlot)
# Actors getter
assert len(pane1.actors) == 2
assert len(pane2.actors) == 2
assert pane1.actors[0] is not pane2.actors[0]
# Actors add
pane1.add_actors(pane2.actors)
assert len(pane1.actors) == 4
assert pane1.actors[3] is pane2.actors[1]
# Actors remove
save_actor = pane1.actors[0]
pane1.remove_actors([pane1.actors[0]])
assert pane1.actors[2] is pane2.actors[1]
# Actors remove all
pane1.add_actors([save_actor])
assert len(pane1.actors) == 4
pane1.remove_all_actors()
assert len(pane1.actors) == 0
# Connect camera
save_vtk_camera2 = pane2.vtk_camera
assert pane1.vtk_camera is not save_vtk_camera2
pane1.link_camera(pane2)
assert pane1.vtk_camera is save_vtk_camera2
# Unconnect camera
pane2.unlink_camera()
assert pane2.vtk_camera is not save_vtk_camera2
# SetBackground
pane1.set_background(0, 0, 0)
assert list(renWin1.GetRenderers())[0].GetBackground() == (0, 0, 0)
# Cleanup
pane1._cleanup(model1)
pane2._cleanup(model2)
def test_vtk_pane_from_renwin(document, comm):
renWin = make_render_window()
pane = VTK(renWin)
# Create pane
model = pane.get_root(document, comm=comm)
assert isinstance(model, VTKSynchronizedPlot)
assert pane._models[model.ref['id']][0] is model
# Check array release when actor are removed from scene
ctx = pane._contexts[model.id]
assert len(ctx.dataArrayCache.keys()) == 5
pane.remove_all_actors()
# Default : 20s before removing arrays
assert len(ctx.dataArrayCache.keys()) == 5
# Force 0s for removing arrays
ctx.checkForArraysToRelease(0)
assert len(ctx.dataArrayCache.keys()) == 0
# Cleanup
pane._cleanup(model)
assert pane._contexts == {}
assert pane._models == {}