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()