Add test code and change negative number detect

Inception95 · Inception95 · commit d0d64ffa2d97 · 2019-12-13T23:13:29.000-08:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -2921,9 +2921,8 @@ def pie(self, x, explode=None, labels=None, colors=None,
                 "and will be removed %(removal)s; pass a 1D array instead.")
             x = np.atleast_1d(x.squeeze())
 
-        for part in x:
-            if part < 0:
-                raise ValueError("wedge sizes must be non negative values")
+        if np.any(x < 0):
+            raise ValueError("wedge sizes must be non negative values")
 
         sx = x.sum()
         if sx > 1:
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -6849,3 +6849,10 @@ def test_bbox_aspect_axes_init():
         sizes.extend([bb.width, bb.height])
 
     assert_allclose(sizes, sizes[0])
+
+
+def test_pi_get_negative_values():
+    # Test the ValueError raised when feeding negative values into axes.pie
+    fig, ax = plt.subplots()
+    with pytest.raises(ValueError):
+        ax.pie([5, 5, -3], explode=[0, .1, .2])
