def test_get_3d_backend():
"""Test get_3d_backend function call for side-effects."""
from mne.viz.backends import renderer
# Test twice to ensure the first call had no side-effect
orig_backend = renderer.MNE_3D_BACKEND
assert renderer.get_3d_backend() == orig_backend
assert renderer.get_3d_backend() == orig_backend
def test_backend_environment_setup(backend, backend_mocker, monkeypatch):
"""Test set up 3d backend based on env."""
monkeypatch.setenv("MNE_3D_BACKEND", backend)
assert os.environ['MNE_3D_BACKEND'] == backend # just double-check
# reload the renderer to check if the 3d backend selection by
# environment variable has been updated correctly
from mne.viz.backends import renderer
importlib.reload(renderer)
assert renderer.MNE_3D_BACKEND == backend
assert get_3d_backend() == backend
def test_renderer(renderer, monkeypatch):
"""Test that renderers are available on demand."""
backend = renderer.get_3d_backend()
cmd = [
sys.executable, '-uc',
'import mne; mne.viz.create_3d_figure((800, 600), show=True); '
'backend = mne.viz.get_3d_backend(); '
'assert backend == %r, backend' % (backend, )
]
monkeypatch.setenv('MNE_3D_BACKEND', backend)
run_subprocess(cmd)
def test_backend_environment_setup(backend, backend_mocker, monkeypatch):
"""Test set up 3d backend based on env."""
monkeypatch.setenv("MNE_3D_BACKEND", backend)
assert os.environ['MNE_3D_BACKEND'] == backend # just double-check
# reload the renderer to check if the 3d backend selection by
# environment variable has been updated correctly
from mne.viz.backends import renderer
importlib.reload(renderer)
assert renderer.MNE_3D_BACKEND == backend
assert get_3d_backend() == backend
def test_3d_backend(renderer):
"""Test default plot."""
# set data
win_size = (600, 600)
win_color = (0, 0, 0)
tet_size = 1.0
tet_x = np.array([0, tet_size, 0, 0])
tet_y = np.array([0, 0, tet_size, 0])
tet_z = np.array([0, 0, 0, tet_size])
tet_indices = np.array([[0, 1, 2], [0, 1, 3], [0, 2, 3], [1, 2, 3]])
tet_color = (1, 1, 1)
sph_center = np.column_stack((tet_x, tet_y, tet_z))
sph_color = (1, 0, 0)
sph_scale = tet_size / 3.0
ct_scalars = np.array([0.0, 0.0, 0.0, 1.0])
ct_levels = [0.2, 0.4, 0.6, 0.8]
ct_surface = {"rr": sph_center, "tris": tet_indices}
qv_mode = "arrow"
qv_color = (0, 0, 1)
qv_scale = tet_size / 2.0
qv_center = np.array([
np.mean((sph_center[va, :], sph_center[vb, :], sph_center[vc, :]),
axis=0) for (va, vb, vc) in tet_indices
])
center = np.mean(qv_center, axis=0)
qv_dir = qv_center - center
qv_scale_mode = 'scalar'
qv_scalars = np.linspace(1.0, 2.0, 4)
txt_x = 0.0
txt_y = 0.0
txt_text = "renderer"
txt_width = 1.0
cam_distance = 5 * tet_size
# init scene
rend = renderer._Renderer(size=win_size, bgcolor=win_color)
rend.set_interactive()
# use mesh
rend.mesh(x=tet_x,
y=tet_y,
z=tet_z,
triangles=tet_indices,
color=tet_color)
# use contour
rend.contour(surface=ct_surface, scalars=ct_scalars, contours=ct_levels)
# use sphere
rend.sphere(center=sph_center, color=sph_color, scale=sph_scale)
# use quiver3d
rend.quiver3d(x=qv_center[:, 0],
y=qv_center[:, 1],
z=qv_center[:, 2],
u=qv_dir[:, 0],
v=qv_dir[:, 1],
w=qv_dir[:, 2],
color=qv_color,
scale=qv_scale,
scale_mode=qv_scale_mode,
scalars=qv_scalars,
mode=qv_mode)
# use tube
rend.tube(origin=np.array([[0, 0, 0]]), destination=np.array([[0, 1, 0]]))
rend.tube(origin=np.array([[1, 0, 0]]),
destination=np.array([[1, 1, 0]]),
scalars=np.array([[1.0, 1.0]]))
# use text
rend.text2d(x=txt_x, y=txt_y, text=txt_text, width=txt_width)
if renderer.get_3d_backend() == "pyvista":
with pytest.raises(NotImplementedError):
rend.text3d(x=0, y=0, z=0, text=txt_text, scale=1.0)
else:
rend.text3d(x=0, y=0, z=0, text=txt_text, scale=1.0)
rend.set_camera(azimuth=180.0,
elevation=90.0,
distance=cam_distance,
focalpoint=center)
rend.show()
def test_3d_backend():
"""Test 3d backend degenerate scenarios and default plot."""
pytest.raises(ValueError, set_3d_backend, "unknown_backend")
pytest.raises(TypeError, set_3d_backend, 1)
assert get_3d_backend() == "mayavi"
# smoke test
set_3d_backend('mayavi')
set_3d_backend('mayavi')
# set data
win_size = (600, 600)
win_color = (0, 0, 0)
tet_size = 1.0
tet_x = np.array([0, tet_size, 0, 0])
tet_y = np.array([0, 0, tet_size, 0])
tet_z = np.array([0, 0, 0, tet_size])
tet_indices = np.array([[0, 1, 2],
[0, 1, 3],
[0, 2, 3],
[1, 2, 3]])
tet_color = (1, 1, 1)
sph_center = np.column_stack((tet_x, tet_y, tet_z))
sph_color = (1, 0, 0)
sph_scale = tet_size / 3.0
qv_mode = "arrow"
qv_color = (0, 0, 1)
qv_scale = tet_size / 2.0
qv_center = np.array([np.mean((sph_center[va, :],
sph_center[vb, :],
sph_center[vc, :]), axis=0)
for (va, vb, vc) in tet_indices])
center = np.mean(qv_center, axis=0)
qv_dir = qv_center - center
txt_x = 0.0
txt_y = 0.0
txt_text = "renderer"
txt_width = 1.0
cam_distance = 5 * tet_size
# init scene
renderer = _Renderer(size=win_size, bgcolor=win_color)
renderer.set_interactive()
# use mesh
renderer.mesh(x=tet_x, y=tet_y, z=tet_z,
triangles=tet_indices,
color=tet_color)
# use sphere
renderer.sphere(center=sph_center, color=sph_color,
scale=sph_scale)
# use quiver3d
renderer.quiver3d(x=qv_center[:, 0],
y=qv_center[:, 1],
z=qv_center[:, 2],
u=qv_dir[:, 0],
v=qv_dir[:, 1],
w=qv_dir[:, 2],
color=qv_color,
scale=qv_scale,
mode=qv_mode)
# use text
renderer.text(x=txt_x, y=txt_y, text=txt_text, width=txt_width)
renderer.set_camera(azimuth=180.0, elevation=90.0, distance=cam_distance,
focalpoint=center)
renderer.show()
