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
|
# Author: Suyog Dutt Jain <suyog.jain@aero.iitb.ac.in>
# Prabhu Ramachandran <prabhu_r@users.sf.net>
# Copyright (c) 2008-2020, Enthought, Inc.
# License: BSD Style.
# Standard library imports.
from os.path import abspath
from io import BytesIO
import copy
import unittest
# Local imports.
from mayavi.tests.common import get_example_data
# Enthought library imports
from mayavi.core.null_engine import NullEngine
from mayavi.sources.plot3d_reader import PLOT3DReader
from mayavi.filters.select_output import SelectOutput
from mayavi.modules.outline import Outline
class TestPlot3dMbReader(unittest.TestCase):
def setUp(self):
e = NullEngine()
# Uncomment to see visualization for debugging etc.
# e = Engine()
e.start()
e.new_scene()
self.e = e
# Read the multi-block plot3d file.
r = PLOT3DReader()
r.reader.trait_set(has_byte_count=True, multi_grid=True,
byte_order='little_endian')
r.initialize(get_example_data('tiny.xyz'),
get_example_data('tiny.q'),
configure=False)
e.add_source(r)
# Add the filter.
f = SelectOutput()
e.add_filter(f)
# Create an outline for the data.
o = Outline()
e.add_module(o)
o.render()
self.o = o
self.r = r
self.e = e
self.scene = e.current_scene
return
def tearDown(self):
self.e.stop()
return
def test_plot3d_mb_reader(self):
o = self.o
# Check the bounds of the outline.
self.assertEqual(o.outline_filter.output.bounds,
(1.0, 2.0, 1.0, 2.0, 1.0, 2.0))
# from mayavi.tools.show import show
# show()
def test_save_and_restore(self):
engine = self.e
scene = self.scene
# Save visualization.
f = BytesIO()
f.name = abspath('test.mv2') # We simulate a file.
engine.save_visualization(f)
f.seek(0) # So we can read this saved data.
# Remove existing scene.
engine.close_scene(scene)
# Load visualization
engine.load_visualization(f)
s = self.scene = engine.current_scene
o = s.children[0].children[0].children[0].children[0]
self.assertEqual(o.outline_filter.output.bounds,
(1.0, 2.0, 1.0, 2.0, 1.0, 2.0))
def test_deepcopied(self):
############################################################
# Copy the reader to see if it does not pop up the UI.
# Pop the source object.
r = self.r
e = self.e
r1 = copy.deepcopy(r)
e.add_source(r1)
o1 = r1.children[0].children[0].children[0]
self.assertEqual(o1.outline_filter.output.bounds,
(1.0, 2.0, 1.0, 2.0, 1.0, 2.0))
r1.children[0].output_index = 1
self.assertEqual(o1.outline_filter.output.bounds,
(2.0, 3.0, 1.0, 2.0, 1.0, 2.0))
if __name__ == '__main__':
unittest.main()
|
