def test_layered_mesh(renderer_interactive):
"""Test management of scalars/colormap overlay."""
if renderer_interactive._get_3d_backend() != 'pyvista':
pytest.skip('TimeViewer tests only supported on PyVista')
mesh = _LayeredMesh(
renderer=renderer_interactive._get_renderer(size=(300, 300)),
vertices=np.array([[0, 0, 0], [1, 0, 0], [0, 1, 0], [1, 1, 0]]),
triangles=np.array([[0, 1, 2], [1, 2, 3]]),
normals=np.array([[0, 0, 1]] * 4),
)
assert not mesh._is_mapped
mesh.map()
assert mesh._is_mapped
assert mesh._cache is None
mesh.update()
assert len(mesh._overlays) == 0
mesh.add_overlay(
scalars=np.array([0, 1, 1, 0]),
colormap=np.array([(1, 1, 1, 1), (0, 0, 0, 0)]),
rng=[0, 1],
opacity=None,
name='test',
)
assert mesh._cache is not None
assert len(mesh._overlays) == 1
assert 'test' in mesh._overlays
mesh.remove_overlay('test')
assert len(mesh._overlays) == 0
mesh._clean()
def test_layered_mesh(renderer_interactive_pyvistaqt):
"""Test management of scalars/colormap overlay."""
mesh = _LayeredMesh(
renderer=renderer_interactive_pyvistaqt._get_renderer(size=(300, 300)),
vertices=np.array([[0, 0, 0], [1, 0, 0], [0, 1, 0], [1, 1, 0]]),
triangles=np.array([[0, 1, 2], [1, 2, 3]]),
normals=np.array([[0, 0, 1]] * 4),
)
assert not mesh._is_mapped
mesh.map()
assert mesh._is_mapped
assert mesh._current_colors is None
assert mesh._cached_colors is None
mesh.update()
assert len(mesh._overlays) == 0
mesh.add_overlay(
scalars=np.array([0, 1, 1, 0]),
colormap=np.array([(1, 1, 1, 1), (0, 0, 0, 0)]),
rng=[0, 1],
opacity=None,
name='test1',
)
assert mesh._current_colors is not None
assert mesh._cached_colors is None
assert len(mesh._overlays) == 1
assert 'test1' in mesh._overlays
mesh.add_overlay(
scalars=np.array([1, 0, 0, 1]),
colormap=np.array([(1, 1, 1, 1), (0, 0, 0, 0)]),
rng=[0, 1],
opacity=None,
name='test2',
)
assert mesh._current_colors is not None
assert mesh._cached_colors is not None
assert len(mesh._overlays) == 2
assert 'test2' in mesh._overlays
mesh.remove_overlay('test2')
assert 'test2' not in mesh._overlays
mesh.update()
assert len(mesh._overlays) == 1
mesh._clean()