def test_vtkjs_pane(document, comm, tmp_path):
# from url
url = r'https://raw.githubusercontent.com/Kitware/vtk-js/master/Data/StanfordDragon.vtkjs'
pane_from_url = VTK(url)
# Create pane
model = pane_from_url.get_root(document, comm=comm)
assert isinstance(model, VTKJSPlot)
assert pane_from_url._models[model.ref['id']][0] is model
assert isinstance(model.data, str)
with BytesIO(base64.b64decode(model.data.encode())) as in_memory:
with ZipFile(in_memory) as zf:
filenames = zf.namelist()
assert len(filenames) == 9
assert 'StanfordDragon.obj/index.json' in filenames
# Export Update and Read
tmpfile = os.path.join(*tmp_path.joinpath('export.vtkjs').parts)
pane_from_url.export_vtkjs(filename=tmpfile)
with open(tmpfile, 'rb') as file_exported:
pane_from_url.object = file_exported
#test from file
pane_from_file = VTK(tmpfile)
model_from_file = pane_from_file.get_root(document, comm=comm)
assert isinstance(pane_from_file, VTKJS)
assert isinstance(model_from_file, VTKJSPlot)
def test_vtk_pane_from_renwin(document, comm, tmp_path):
renWin = make_render_window()
pane = VTK(renWin)
# Create pane
model = pane.get_root(document, comm=comm)
assert isinstance(model, VTKPlot)
assert pane._models[model.ref['id']][0] is model
with BytesIO(base64.b64decode(model.data.encode())) as in_memory:
with ZipFile(in_memory) as zf:
filenames = zf.namelist()
assert len(filenames) == 4
assert 'index.json' in filenames
# Export Update and Read
tmpfile = os.path.join(*tmp_path.joinpath('export.vtkjs').parts)
pane.export_vtkjs(filename=tmpfile)
with open(tmpfile, 'rb') as file_exported:
pane.object = file_exported
# Cleanup
pane._cleanup(model)
assert pane._models == {}