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
|
from __future__ import annotations
import pytest
import pyvista as pv
from pyvista import _vtk
KEY = 'Data'
@pytest.fixture
def scalar_bars(sphere):
sphere[KEY] = sphere.points[:, 2]
plotter = pv.Plotter()
plotter.add_mesh(sphere, show_scalar_bar=False)
plotter.add_scalar_bar(
KEY,
interactive=True,
vertical=False,
outline=True,
fmt='%10.5f',
nan_annotation=True,
fill=True,
background_color='k',
)
return plotter.scalar_bars
def test_repr(scalar_bars):
repr_ = repr(scalar_bars)
assert f'"{KEY}"' in repr_
assert 'False' in repr_ or 'True' in repr_, 'missing interactive flag'
assert 'Scalar Bar Title Interactive' in repr_
def test_remove_fail(scalar_bars):
scalar_bars.add_scalar_bar('MOARDATA', mapper=scalar_bars._plotter.mapper)
with pytest.raises(ValueError, match='Multiple scalar bars'):
scalar_bars.remove_scalar_bar()
def test_add_fail(scalar_bars):
with pytest.raises(ValueError, match='Mapper cannot be ``None``'):
scalar_bars.add_scalar_bar('MOARDATA')
def test_dict(scalar_bars):
assert KEY in scalar_bars
assert 'Data' in scalar_bars.keys()
assert len(scalar_bars) == 1
assert next(iter(scalar_bars.keys())) == KEY
assert isinstance(next(iter(scalar_bars.values())), _vtk.vtkScalarBarActor)
for key, value in scalar_bars.items():
assert isinstance(value, _vtk.vtkScalarBarActor)
assert key == 'Data'
assert isinstance(scalar_bars[KEY], _vtk.vtkScalarBarActor)
def test_clear(scalar_bars):
assert len(scalar_bars) == 1
scalar_bars.clear()
assert len(scalar_bars) == 0
def test_too_many_scalar_bars():
pl = pv.Plotter()
with pytest.raises(RuntimeError, match='Maximum number of color'): # noqa: PT012
for i in range(100):
mesh = pv.Sphere()
mesh[str(i)] = range(mesh.n_points)
pl.add_mesh(mesh)
|
