Merge pull request #7525 from rmjarvis/quiver-nohead

efiring · web-flow · commit b9c3b6400bd1 · 2016-11-28T09:08:33.000-10:00
[MRG+1] Avoid division by zero if headlength=0 for quiver
diff --git a/lib/matplotlib/quiver.py b/lib/matplotlib/quiver.py
@@ -702,7 +702,7 @@ def _h_arrows(self, length):
         X0 = x0.take(ii)
         Y0 = y0.take(ii)
         Y0[3:-1] *= -1
-        shrink = length / minsh
+        shrink = length / minsh if minsh != 0. else 0.
         X0 = shrink * X0[np.newaxis, :]
         Y0 = shrink * Y0[np.newaxis, :]
         short = np.repeat(length < minsh, 8, axis=1)
diff --git a/lib/matplotlib/tests/test_quiver.py b/lib/matplotlib/tests/test_quiver.py
@@ -59,6 +59,19 @@ def test_no_warnings():
     assert len(w) == 0
 
 
+@cleanup
+def test_zero_headlength():
+    # Based on report by Doug McNeil:
+    # http://matplotlib.1069221.n5.nabble.com/quiver-warnings-td28107.html
+    fig, ax = plt.subplots()
+    X, Y = np.meshgrid(np.arange(10), np.arange(10))
+    U, V = np.cos(X), np.sin(Y)
+    with warnings.catch_warnings(record=True) as w:
+        ax.quiver(U, V, headlength=0, headaxislength=0)
+        fig.canvas.draw()
+    assert len(w) == 0
+
+
 @image_comparison(baseline_images=['quiver_animated_test_image'],
                   extensions=['png'])
 def test_quiver_animate():
