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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
"""This conftest is here to allow for checking garbage collection and
memory leaks for all plotting tests
"""
from __future__ import annotations
import gc
import inspect
import pytest
import pyvista as pv
from pyvista.plotting import system_supports_plotting
# these are set here because we only need them for plotting tests
pv.OFF_SCREEN = True
SKIP_PLOTTING = not system_supports_plotting()
# Configure skip_plotting marker
def pytest_configure(config):
config.addinivalue_line(
'markers',
'skip_plotting: skip the test if system does not support plotting',
)
def pytest_runtest_setup(item):
skip = any(mark.name == 'skip_plotting' for mark in item.iter_markers())
if skip and SKIP_PLOTTING:
pytest.skip('Test requires system to support plotting')
def _is_vtk(obj):
try:
return obj.__class__.__name__.startswith('vtk')
except (ReferenceError, AttributeError):
return False
@pytest.fixture(autouse=True)
def check_gc(request):
"""Ensure that all VTK objects are garbage-collected by Python."""
if request.node.get_closest_marker('skip_check_gc'):
yield
return
gc.collect()
before = {id(o) for o in gc.get_objects() if _is_vtk(o)}
yield
pv.close_all()
gc.collect()
after = [o for o in gc.get_objects() if _is_vtk(o) and id(o) not in before]
msg = 'Not all objects GCed:\n'
for obj in after:
cn = obj.__class__.__name__
cf = inspect.currentframe()
referrers = [v for v in gc.get_referrers(obj) if v is not after and v is not cf]
del cf
for ri, referrer in enumerate(referrers):
if isinstance(referrer, dict):
for k, v in referrer.items():
if k is obj:
referrers[ri] = 'dict: d key'
del k, v
break
elif v is obj:
referrers[ri] = f'dict: d[{k!r}]'
del k, v
break
del k, v
else:
referrers[ri] = f'dict: len={len(referrer)}'
else:
referrers[ri] = repr(referrer)
del ri, referrer
msg += f'{cn} at {hex(id(obj))}: {referrers}\n'
del cn, referrers
assert len(after) == 0, msg
@pytest.fixture
def colorful_tetrahedron():
mesh = pv.Tetrahedron()
mesh.cell_data['colors'] = [[255, 255, 255], [255, 0, 0], [0, 255, 0], [0, 0, 255]]
return mesh
@pytest.fixture(autouse=True)
def set_default_theme():
"""Reset the testing theme for every test."""
pv.global_theme.load_theme(pv.plotting.themes._TestingTheme())
yield
pv.global_theme.load_theme(pv.plotting.themes._TestingTheme())
def make_two_char_img(text):
"""Turn text into an image.
This is really only here to make a two character black and white image.
"""
# create a basic texture by plotting a sphere and converting the image
# buffer to a texture
pl = pv.Plotter(window_size=(300, 300), lighting=None, off_screen=True)
pl.add_text(text, color='w', font_size=100, position=(0.1, 0.1), viewport=True, font='courier')
pl.background_color = 'k'
pl.camera.zoom = 'tight'
return pv.Texture(pl.screenshot()).to_image()
@pytest.fixture
def cubemap():
"""Sample texture as a cubemap."""
return pv.Texture(
[
make_two_char_img('X+'),
make_two_char_img('X-'),
make_two_char_img('Y+'),
make_two_char_img('Y-'),
make_two_char_img('Z+'),
make_two_char_img('Z-'),
],
)
|
