def test_vtk_pane_more_complex(document, comm, tmp_path):
renWin = pyvista_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
colorbars_infered = pane.construct_colorbars().object
assert len(colorbars_infered.below) == 2 # infer only actor color bars
assert all(isinstance(cb, ColorBar) for cb in colorbars_infered.below)
colorbars_in_scene = pane.construct_colorbars(infer=False).object()
assert len(colorbars_in_scene.below) == 3
assert all(isinstance(cb, ColorBar) for cb in colorbars_in_scene.below)
# add axes
pane.axes = dict(
origin=[-5, 5, -2],
xticker={'ticks': np.linspace(-5, 5, 5)},
yticker={'ticks': np.linspace(-5, 5, 5)},
zticker={
'ticks': np.linspace(-2, 2, 5),
'labels':
[''] + [str(int(item)) for item in np.linspace(-2, 2, 5)[1:]]
},
fontsize=12,
digits=1,
grid_opacity=0.5,
show_grid=True)
assert isinstance(model.axes, VTKAxes)
# test export to file
tmpfile = os.path.join(*tmp_path.joinpath('scene').parts)
exported_file = pane.export_scene(filename=tmpfile)
assert exported_file.endswith('.synch')
# test import from file
# (TODO test if the scene imported is identical to the one exported)
imported_scene = VTK.import_scene(filename=exported_file)
assert isinstance(imported_scene, VTKRenderWindowSynchronized)
# Cleanup
pane._cleanup(model)
assert pane._contexts == {}
assert pane._models == {}
def test_vtk_serialize_on_instantiation(document, comm, tmp_path):
renWin = make_render_window()
pane = VTK(renWin, serialize_on_instantiation=True)
assert isinstance(pane, VTKRenderWindow)
model = pane.get_root(document, comm=comm)
assert isinstance(model, VTKSynchronizedPlot)
pane.param.trigger('object')
# test export to file
tmpfile = os.path.join(*tmp_path.joinpath('scene').parts)
exported_file = pane.export_scene(filename=tmpfile)
assert exported_file.endswith('.synch')
# test import from file
imported_pane = VTK.import_scene(filename=exported_file,
synchronizable=False)
assert isinstance(imported_pane, VTKRenderWindow)