Add legend to acorr image test

dstansby · dstansby · commit cc3a708398a0 · 2017-12-06T19:01:09.000Z
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -1760,7 +1760,7 @@ def xcorr(self, x, y, normed=True, detrend=mlab.detrend_none,
         if usevlines:
             a = self.vlines(lags, [0], correls, **kwargs)
             # Make label empty so only vertical lines get a legend entry
-            kwargs['label'] = ''
+            kwargs.pop('label', '')
             b = self.axhline(**kwargs)
         else:
             kwargs.setdefault('marker', 'o')
diff --git a/lib/matplotlib/tests/baseline_images/test_axes/acorr.png b/lib/matplotlib/tests/baseline_images/test_axes/acorr.png
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -51,7 +51,8 @@ def test_acorr():
     x = np.random.normal(0, 1, n).cumsum()
 
     fig, ax = plt.subplots()
-    ax.acorr(x, maxlags=n - 1)
+    ax.acorr(x, maxlags=n - 1, label='acorr')
+    ax.legend()
 
 
 @image_comparison(baseline_images=['spy'], extensions=['png'], style='mpl20')
