forked from pyvista/pyvista
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_export.py
96 lines (83 loc) · 3.29 KB
/
test_export.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
import os
import sys
import numpy as np
import pytest
import vtk
import vtki
from vtki import examples as ex
from vtki.plotting import running_xserver
if __name__ != '__main__':
OFF_SCREEN = 'pytest' in sys.modules
else:
OFF_SCREEN = False
@pytest.mark.skipif(not running_xserver(), reason="Requires X11")
def test_export_single(tmpdir):
filename = str(tmpdir.mkdir("tmpdir").join('scene'))
data = ex.load_airplane()
# Create the scene
plotter = vtki.Plotter(off_screen=OFF_SCREEN)
plotter.add_mesh(data)
plotter.export_vtkjs(filename)
cpos_out = plotter.show() # Export must be called before showing!
plotter.close()
# Now make sure the file is there
assert os.path.isfile('{}.vtkjs'.format(filename))
@pytest.mark.skipif(not running_xserver(), reason="Requires X11")
def test_export_multi(tmpdir):
filename = str(tmpdir.mkdir("tmpdir").join('scene'))
multi = vtki.MultiBlock()
# Add examples
multi.append(ex.load_ant())
multi.append(ex.load_sphere())
multi.append(ex.load_uniform())
multi.append(ex.load_airplane())
multi.append(ex.load_rectilinear())
# Create the scene
plotter = vtki.Plotter(off_screen=OFF_SCREEN)
plotter.add_mesh(multi)
plotter.export_vtkjs(filename, compress_arrays=True)
cpos_out = plotter.show() # Export must be called before showing!
plotter.close()
# Now make sure the file is there
assert os.path.isfile('{}.vtkjs'.format(filename))
@pytest.mark.skipif(not running_xserver(), reason="Requires X11")
def test_export_texture(tmpdir):
filename = str(tmpdir.mkdir("tmpdir").join('scene'))
data = ex.load_globe()
# Create the scene
plotter = vtki.Plotter(off_screen=OFF_SCREEN)
plotter.add_mesh(data, texture=True)
plotter.export_vtkjs(filename)
cpos_out = plotter.show() # Export must be called before showing!
plotter.close()
# Now make sure the file is there
assert os.path.isfile('{}.vtkjs'.format(filename))
@pytest.mark.skipif(not running_xserver(), reason="Requires X11")
def test_export_verts(tmpdir):
filename = str(tmpdir.mkdir("tmpdir").join('scene'))
data = vtki.PolyData(np.random.rand(100, 3))
# Create the scene
plotter = vtki.Plotter(off_screen=OFF_SCREEN)
plotter.add_mesh(data)
plotter.export_vtkjs(filename)
cpos_out = plotter.show() # Export must be called before showing!
plotter.close()
# Now make sure the file is there
assert os.path.isfile('{}.vtkjs'.format(filename))
@pytest.mark.skipif(not running_xserver(), reason="Requires X11")
def test_export_color(tmpdir):
filename = str(tmpdir.mkdir("tmpdir").join('scene'))
data = ex.load_airplane()
# Create the scene
plotter = vtki.Plotter(off_screen=OFF_SCREEN)
plotter.add_mesh(data, color='yellow')
plotter.export_vtkjs(filename)
cpos_out = plotter.show() # Export must be called before showing!
plotter.close()
# Now make sure the file is there
assert os.path.isfile('{}.vtkjs'.format(filename))
def test_vtkjs_url():
file_url = 'https://www.dropbox.com/s/6m5ttdbv5bf4ngj/ripple.vtkjs?dl=0'
vtkjs_url = 'http://viewer.vtki.org/?fileURL=https://dl.dropbox.com/s/6m5ttdbv5bf4ngj/ripple.vtkjs?dl=0'
assert vtkjs_url in vtki.get_vtkjs_url(file_url)
assert vtkjs_url in vtki.get_vtkjs_url('dropbox', file_url)