MAINT np.testing.assert_raises depends on nose

NelleV · NelleV · commit ff71b3632fc7 · 2017-01-22T14:19:30.000-08:00
diff --git a/lib/matplotlib/tests/test_gridspec.py b/lib/matplotlib/tests/test_gridspec.py
@@ -1,5 +1,6 @@
 import matplotlib.gridspec as gridspec
-from numpy.testing import assert_raises, assert_equal
+from numpy.testing import assert_equal
+import pytest
 
 
 def test_equal():
@@ -13,14 +14,14 @@ def test_width_ratios():
     Addresses issue #5835.
     See at https://github.com/matplotlib/matplotlib/issues/5835.
     """
-    assert_raises(ValueError, gridspec.GridSpec,
-                  1, 1, width_ratios=[2, 1, 3])
+    with pytest.raises(ValueError):
+        gridspec.GridSpec(1, 1, width_ratios=[2, 1, 3])
 
 
 def test_height_ratios():
     """
     Addresses issue #5835.
     See at https://github.com/matplotlib/matplotlib/issues/5835.
     """
-    assert_raises(ValueError, gridspec.GridSpec,
-                  1, 1, height_ratios=[2, 1, 3])
+    with pytest.raises(ValueError):
+        gridspec.GridSpec(1, 1, height_ratios=[2, 1, 3])
diff --git a/lib/matplotlib/tests/test_image.py b/lib/matplotlib/tests/test_image.py
@@ -9,6 +9,7 @@
 from nose.plugins.attrib import attr
 
 import numpy as np
+from numpy.testing import assert_array_equal
 
 from matplotlib.testing.decorators import (image_comparison,
                                            knownfailureif, cleanup)
