Added tests for invalid axes limits

Vedant Nanda · Vedant Nanda · commit b4fa6d151131 · 2017-03-04T22:11:09.000+05:30
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -4974,23 +4974,30 @@ def test_bar_single_height():
     ax.bar(0, 1, bottom=range(4), width=1, orientation='horizontal')
 
 
-def invalid_axes_limits():
+def test_invalid_axes_limits():
+    error_string = 'NaN or Inf cannot be the argument values'
     with pytest.raises(ValueError) as err:
-        plt.set_xlim(left=np.nan)
+        plt.xlim(left=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_xlim(left=np.inf)
+        plt.xlim(left=np.inf)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_xlim(right=np.nan)
+        plt.xlim(right=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_xlim(right=np.inf)
+        plt.xlim(right=np.inf)
+    assert err.value.message == error_string
 
     with pytest.raises(ValueError) as err:
-        plt.set_ylim(bottom=np.nan)
+        plt.ylim(bottom=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_ylim(bottom=np.inf)
+        plt.ylim(bottom=np.inf)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_ylim(top=np.nan)
+        plt.ylim(top=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_ylim(top=np.inf)
-
-    err.match('NaN or Inf cannot be the argument values')
+        plt.ylim(top=np.inf)
+    assert err.value.message == error_string
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -493,32 +493,45 @@ def test_autoscale():
     assert ax.get_w_lims() == (0, 1, -.1, 1.1, -.4, 2.4)
 
 
-def invalid_axes_limits():
+def test_invalid_axes_limits():
+    error_string = 'NaN or Inf cannot be the argument values'
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
     with pytest.raises(ValueError) as err:
-        plt.set_xlim3d(left=np.nan)
+        ax.set_xlim3d(left=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_xlim3d(left=np.inf)
+        ax.set_xlim3d(left=np.inf)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_xlim3d(right=np.nan)
+        ax.set_xlim3d(right=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_xlim3d(right=np.inf)
+        ax.set_xlim3d(right=np.inf)
+    assert err.value.message == error_string
 
     with pytest.raises(ValueError) as err:
-        plt.set_ylim3d(bottom=np.nan)
+        ax.set_ylim3d(bottom=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_ylim3d(bottom=np.inf)
+        ax.set_ylim3d(bottom=np.inf)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_ylim3d(top=np.nan)
+        ax.set_ylim3d(top=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_ylim3d(top=np.inf)
+        ax.set_ylim3d(top=np.inf)
+    assert err.value.message == error_string
 
     with pytest.raises(ValueError) as err:
-        plt.set_zlim3d(bottom=np.nan)
+        ax.set_zlim3d(bottom=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_zlim3d(bottom=np.inf)
+        ax.set_zlim3d(bottom=np.inf)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_zlim3d(top=np.nan)
+        ax.set_zlim3d(top=np.nan)
+    assert err.value.message == error_string
     with pytest.raises(ValueError) as err:
-        plt.set_zlim3d(top=np.inf)
-
-    err.match('NaN or Inf cannot be the argument values')
+        ax.set_zlim3d(top=np.inf)
+    assert err.value.message == error_string
