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