def test_threshold_filter_nan(self):
src = self.make_src(nan=True)
self.e.add_source(src)
threshold = Threshold()
self.e.add_filter(threshold)
self.assertEqual(
np.nanmin(src.scalar_data),
np.nanmin(
threshold.get_output_dataset().point_data.scalars.to_array()))
self.assertEqual(
np.nanmax(src.scalar_data),
np.nanmax(
threshold.get_output_dataset().point_data.scalars.to_array()))
def test_threshold_with_other_filter_as_input(self):
# Given
x, y, z = np.mgrid[-1:1:10j, -1:1:10j, -1:1:10j]
s = x * x + y * y + z * z
src = ArraySource(scalar_data=s)
self.e.add_source(src)
scp = CutPlane()
self.e.add_filter(scp)
# When
threshold = Threshold()
self.e.add_filter(threshold)
threshold.set(lower_threshold=0.25,
upper_threshold=0.75,
auto_reset_lower=False,
auto_reset_upper=False)
# Then
output = threshold.get_output_dataset()
self.assertTrue(output is not None)
self.assertTrue(output.is_a('vtkUnstructuredGrid'))
output_range = output.point_data.scalars.range
self.assertTrue(output_range[0] >= 0.25)
self.assertTrue(output_range[1] <= 0.75)
def test_threshold_with_other_filter_as_input(self):
# Given
x, y, z = np.mgrid[-1:1:10j, -1:1:10j, -1:1:10j]
s = x*x + y*y + z*z
src = ArraySource(scalar_data=s)
self.e.add_source(src)
scp = CutPlane()
self.e.add_filter(scp)
# When
threshold = Threshold()
self.e.add_filter(threshold)
threshold.trait_set(
lower_threshold=0.25, upper_threshold=0.75,
auto_reset_lower=False, auto_reset_upper=False
)
# Then
output = threshold.get_output_dataset()
self.assertTrue(output is not None)
self.assertTrue(output.is_a('vtkUnstructuredGrid'))
output_range = output.point_data.scalars.range
self.assertTrue(output_range[0] >= 0.25)
self.assertTrue(output_range[1] <= 0.75)
def test_threshold_filter_nan(self):
src = self.make_src(nan=True)
self.e.add_source(src)
threshold = Threshold()
self.e.add_filter(threshold)
self.assertEqual(
np.nanmin(src.scalar_data),
np.nanmin(
threshold.get_output_dataset().point_data.scalars.to_array()
)
)
self.assertEqual(
np.nanmax(src.scalar_data),
np.nanmax(
threshold.get_output_dataset().point_data.scalars.to_array()
)
)
def test_threshold_filter_threhsold(self):
src = self.make_src()
self.e.add_source(src)
threshold = Threshold()
self.e.add_filter(threshold)
threshold.upper_threshold = 20.
self.assertTrue(20 >= np.nanmax(
threshold.get_output_dataset().point_data.scalars.to_array()))
return
def test_threshold_filter_threhsold(self):
src = self.make_src()
self.e.add_source(src)
threshold = Threshold()
self.e.add_filter(threshold)
threshold.upper_threshold = 20.
self.assertTrue(
20 >= np.nanmax(
threshold.get_output_dataset().point_data.scalars.to_array()
)
)
return
