def check(self):
"""Do the actual testing"""
scene = self.scene
src = scene.children[0]
self.assertEqual(src.point_scalars_name, 'temperature')
c = src.children[1]
sc = get_output(c.outputs[0]).point_data.scalars
self.assertEqual(sc.name, 'temperature')
# It is an iso-contour!
self.assertEqual(sc.range[0], sc.range[1])
aa = c.children[0].children[0]
self.assertEqual(aa.point_scalars_name, 'pressure')
sc = get_output(aa.outputs[0]).point_data.scalars
self.assertEqual(sc.name, 'pressure')
self.assertEqual((abs(sc.range[0] - 70) < 1.0), True)
self.assertEqual((abs(sc.range[1] - 70) < 1.0), True)
s = aa.children[0].children[0]
def check(self):
"""Do the actual testing"""
scene = self.scene
src = scene.children[0]
self.assertEqual(src.point_scalars_name,'temperature')
c = src.children[1]
sc = get_output(c.outputs[0]).point_data.scalars
self.assertEqual(sc.name,'temperature')
# It is an iso-contour!
self.assertEqual(sc.range[0],sc.range[1])
aa = c.children[0].children[0]
self.assertEqual(aa.point_scalars_name,'pressure')
sc = get_output(aa.outputs[0]).point_data.scalars
self.assertEqual(sc.name, 'pressure')
self.assertEqual((abs(sc.range[0] - 70) < 1.0),True)
self.assertEqual((abs(sc.range[1] - 70) < 1.0),True)
s = aa.children[0].children[0]
def check(self):
"""Check if the visualization is OK. Note that this is not an
image based check, which is very convenient.
"""
e = self.script.engine
scene = e.current_scene
src = scene.children[0]
assert src.point_scalars_name == 'u'
c = src.children[1]
sc = get_output(c.outputs[0]).point_data.scalars
assert sc.name == 'u'
# It is an iso-contour!
assert sc.range[0] == sc.range[1]
aa = c.children[0].children[0]
assert aa.point_scalars_name == 't'
sc = get_output(aa.outputs[0]).point_data.scalars
assert sc.name == 't'
assert abs(sc.range[0] - 308) < 1.0
assert abs(sc.range[1] - 631) < 1.0
s = aa.children[0].children[0]
def check(mod):
"""Check if test status is OK for the given module."""
cut, warper, ctr, normals, a = mod.components
c = ctr.filter
# The intermediate ones are disabled.
assert normals.outputs[0] is cut.outputs[0]
# Enable the contours.
ctr.enabled = True
assert ctr.outputs[0] is c.outputs[0]
assert ctr.outputs[0] is normals.outputs[0]
n_output = get_output(normals.outputs[0])
rng = n_output.point_data.scalars.range
assert (rng[1] - rng[0]) < 1e-4
# Turn on auto-contours
c.auto_contours = True
# Increase number of contours and the range should change.
c.number_of_contours = 10
n_output = get_output(normals.outputs[0])
assert len(n_output.points) != 0
rng = n_output.point_data.scalars.range
assert rng[0] < rng[1]
# Check if pipeline_changed is correctly propagated.
old = get_output(normals.outputs[0])
assert a.mapper.scalar_mode == 'default'
c.filled_contours = True
n_output = get_output(normals.outputs[0])
c_output = get_output(c.outputs[0])
assert n_output != old
assert n_output is c_output
# Check if the actor responds correctly to the
# filled_contour change.
assert a.mapper.scalar_mode == 'use_cell_data'
# Set back everything to original state.
c.filled_contours = False
assert a.mapper.scalar_mode == 'default'
c.number_of_contours = 1
c.auto_contours = False
ctr.enabled = False
assert normals.outputs[0] is cut.outputs[0]
def _setup_vector_data(self):
input = get_output(self.source.outputs[0])
pv = input.point_data.vectors
cv = input.cell_data.vectors
data_attr = DataAttributes(name='No vectors')
point_data_attr = DataAttributes(name='No vectors')
point_data_attr.compute_vector(pv, 'point')
cell_data_attr = DataAttributes(name='No vectors')
cell_data_attr.compute_vector(cv, 'cell')
if self.lut_data_mode == 'auto':
if len(point_data_attr.range) > 0:
data_attr.copy_traits(point_data_attr)
elif len(cell_data_attr.range) > 0:
data_attr.copy_traits(cell_data_attr)
elif self.lut_data_mode == 'point data':
data_attr.copy_traits(point_data_attr)
elif self.lut_data_mode == 'cell data':
data_attr.copy_traits(cell_data_attr)
data_attr.config_lut(self.vector_lut_manager)
def _setup_vector_data(self):
input = get_output(self.source.outputs[0])
pv = input.point_data.vectors
cv = input.cell_data.vectors
data_attr = DataAttributes(name='No vectors')
point_data_attr = DataAttributes(name='No vectors')
point_data_attr.compute_vector(pv, 'point')
cell_data_attr = DataAttributes(name='No vectors')
cell_data_attr.compute_vector(cv, 'cell')
if self.lut_data_mode == 'auto':
if len(point_data_attr.range) > 0:
data_attr.copy_traits(point_data_attr)
elif len(cell_data_attr.range) > 0:
data_attr.copy_traits(cell_data_attr)
elif self.lut_data_mode == 'point data':
data_attr.copy_traits(point_data_attr)
elif self.lut_data_mode == 'cell data':
data_attr.copy_traits(cell_data_attr)
data_attr.config_lut(self.vector_lut_manager)
def _setup_scalar_data(self):
"""Computes the scalar range and an appropriate name for the
lookup table."""
input = get_output(self.source.outputs[0])
ps = input.point_data.scalars
cs = input.cell_data.scalars
data_attr = DataAttributes(name='No scalars')
point_data_attr = DataAttributes(name='No scalars')
point_data_attr.compute_scalar(ps, 'point')
cell_data_attr = DataAttributes(name='No scalars')
cell_data_attr.compute_scalar(cs, 'cell')
if self.lut_data_mode == 'auto':
if len(point_data_attr.range) > 0:
data_attr.copy_traits(point_data_attr)
elif len(cell_data_attr.range) > 0:
data_attr.copy_traits(cell_data_attr)
elif self.lut_data_mode == 'point data':
data_attr.copy_traits(point_data_attr)
elif self.lut_data_mode == 'cell data':
data_attr.copy_traits(cell_data_attr)
data_attr.config_lut(self.scalar_lut_manager)
def _setup_scalar_data(self):
"""Computes the scalar range and an appropriate name for the
lookup table."""
input = get_output(self.source.outputs[0])
ps = input.point_data.scalars
cs = input.cell_data.scalars
data_attr = DataAttributes(name='No scalars')
point_data_attr = DataAttributes(name='No scalars')
point_data_attr.compute_scalar(ps, 'point')
cell_data_attr = DataAttributes(name='No scalars')
cell_data_attr.compute_scalar(cs, 'cell')
if self.lut_data_mode == 'auto':
if len(point_data_attr.range) > 0:
data_attr.copy_traits(point_data_attr)
elif len(cell_data_attr.range) > 0:
data_attr.copy_traits(cell_data_attr)
elif self.lut_data_mode == 'point data':
data_attr.copy_traits(point_data_attr)
elif self.lut_data_mode == 'cell data':
data_attr.copy_traits(cell_data_attr)
data_attr.config_lut(self.scalar_lut_manager)
