Added set_xlim and set_ylim check for non-finite limit values

bcongdon · bcongdon · commit 22387cc983ca · 2017-01-04T19:23:07.000-05:00
diff --git a/doc/users/whats_new/invalid_axes_limits_errors.rst b/doc/users/whats_new/invalid_axes_limits_errors.rst
@@ -0,0 +1,6 @@
+Invalid (Non-finite) Axis Limit Error
+-------------------------------------
+
+When using :func:`set_xlim` and :func:`set_ylim`, passing non-finite values now
+results in a ValueError. The previous behavior resulted in the limits being
+erroneously reset to `(-0.001, 0.001)`.
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -2875,6 +2875,11 @@ def set_xlim(self, left=None, right=None, emit=True, auto=False, **kw):
         if right is not None:
             right = self.convert_xunits(right)
 
+        if ((left is not None and not np.isfinite(left)) or
+                (right is not None and not np.isfinite(right))):
+            raise ValueError("xlim limits must be finite. "
+                             "instead, found: (%s, %s)" % (left, right))
+
         old_left, old_right = self.get_xlim()
         if left is None:
             left = old_left
@@ -3154,6 +3159,11 @@ def set_ylim(self, bottom=None, top=None, emit=True, auto=False, **kw):
         if top is not None:
             top = self.convert_yunits(top)
 
+        if ((top is not None and not np.isfinite(top)) or
+                (bottom is not None and not np.isfinite(bottom))):
+            raise ValueError("ylim limits must be finite. "
+                             "instead, found: (%s, %s)" % (top, bottom))
+
         old_bottom, old_top = self.get_ylim()
 
         if bottom is None:
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -4877,3 +4877,16 @@ def test_scatter_color_masking():
 def test_eventplot_legend():
     plt.eventplot([1.0], label='Label')
     plt.legend()
+
+
+@cleanup
+def test_invalid_axis_limits():
+    plt.plot([0, 1], [0, 1])
+    with pytest.raises(ValueError):
+        plt.xlim(np.nan)
+    with pytest.raises(ValueError):
+        plt.xlim(np.inf)
+    with pytest.raises(ValueError):
+        plt.ylim(np.nan)
+    with pytest.raises(ValueError):
+        plt.ylim(np.inf)
