Add legend to acorr image test

matplotlib · anntzer · Dec 7, 2017 · Dec 6, 2017 · Dec 6, 2017 · Dec 6, 2017
commit cc3a708398a05af96d88dac87d4035d3b848add2
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')