ENH: Ignore known warnings, just to clean up the test suite.

enthought · Apr 11, 2017 · 1d426d4 · 1d426d4
1 parent 89c3967
commit 1d426d4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/chaco/log_mapper.py b/chaco/log_mapper.py
@@ -3,6 +3,7 @@
 # Major library imports
 from numpy import array, isnan, log, log10, exp, zeros, sometrue,\
     floor, ceil, ndarray
+import numpy as np
 
 # Enthought library imports
 from traits.api import Bool, Float
@@ -54,7 +55,8 @@ def map_screen(self, data_array):
             intermediate = data_array*0.0
         else:
             try:
-                mask = (data_array <= LOG_MINIMUM) | isnan(data_array)
+                with np.errstate(invalid='ignore'):
+                    mask = (data_array <= LOG_MINIMUM) | isnan(data_array)
                 if sometrue(mask):
                     data_array = array(data_array, copy=True, ndmin=1)
                     data_array[mask] = self.fill_value

diff --git a/chaco/multi_array_data_source.py b/chaco/multi_array_data_source.py
@@ -1,5 +1,7 @@
 """ Defines the MultiArrayDataSource class.
 """
+import warnings
+
 # Major package imports
 from numpy import nanmax, nanmin, array, shape, ones, bool, newaxis, nan_to_num
 
@@ -179,8 +181,11 @@ def get_bounds(self, value = None, index = None):
                 mini = nanmin(self._data[::, index])
         else:
             # value is None and index is None:
-            maxi = nanmax(self._data)
-            mini = nanmin(self._data)
+            with warnings.catch_warnings():
+                warnings.filterwarnings(
+                    'ignore', "All-NaN (slice|axis) encountered", RuntimeWarning)
+                maxi = nanmax(self._data)
+                mini = nanmin(self._data)
 
         return (mini, maxi)