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
|
"""
Test the mlab null engine.
"""
import unittest
from mayavi import mlab
from mayavi.core.engine import Engine
from mayavi.tools.engine_manager import engine_manager
from mayavi.core.registry import registry
from mayavi.tests.common import get_example_data
################################################################################
# class `TestMlabNullEngineBase`
################################################################################
class TestMlabNullEngineBase(unittest.TestCase):
""" Base class to test mlab with the null engine
"""
def setUp(self):
e = Engine()
e.start()
self._non_null_engine = e
mlab.set_engine(e)
def tearDown(self):
# Check that the NullEngine was not set as the default mlab engine.
if not mlab.get_engine() is self._non_null_engine:
raise AssertionError("The NullEngine has overridden the default one")
engine_manager.current_engine = None
# Unregistering all unused engines.
registry.unregister_engine(self._non_null_engine)
for engine in list(registry.engines):
registry.unregister_engine(engine)
################################################################################
# class `TestRealMlabNullEngine`
################################################################################
class TestRealMlabNullEngine(unittest.TestCase):
"""Tests if the mlab settings via the options.backend and offscreen
options work correctly."""
def setUp(self):
self.backend = mlab.options.backend
def tearDown(self):
mlab.options.backend = self.backend
for engine in list(registry.engines):
registry.unregister_engine(engine)
def test_test_backend_clf(self):
"""Test if setting the backend to 'test' works."""
mlab.options.backend = 'test'
mlab.test_contour3d()
e = mlab.get_engine()
self.assertEqual(len(e.scenes), 1)
self.assertEqual(len(e.scenes[0].children), 1)
mlab.clf()
self.assertEqual(len(e.scenes), 1)
self.assertEqual(len(e.scenes[0].children), 0)
mlab.pipeline.open(get_example_data('cube.vti'))
mlab.clf()
self.assertEqual(len(e.scenes), 1)
self.assertEqual(len(e.scenes[0].children), 0)
def test_points3d_test_backend(self):
# Given
mlab.options.backend = 'test'
# When
g = mlab.test_points3d()
pd = g.actor.mapper.input
# Then
self.assertTrue(pd.number_of_polys == 0)
# When
g.render()
# Then
self.assertTrue(pd.number_of_polys == 1920)
if __name__ == '__main__':
unittest.main()
|
