FIX: minor style change + add @cleanup

tacaswell · tacaswell · commit 2500b2337eb9 · 2017-05-01T00:16:01.000-04:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -2284,10 +2284,14 @@ def handle_single_axis(scale, autoscaleon, shared_axes, interval,
             # ignore non-finite data limits if good limits exist
             finite_dl = [d for d in dl if np.isfinite(d).all()]
             if len(finite_dl):
-                # if finite limits exist for atleast one axis (and the other is infinite), restore the
-                # finite limits
-                x_finite = [d for d in dl if (np.isfinite(d.intervalx).all() and (d not in finite_dl))]
-                y_finite = [d for d in dl if (np.isfinite(d.intervaly).all() and (d not in finite_dl))]
+                # if finite limits exist for atleast one axis (and the
+                # other is infinite), restore the finite limits
+                x_finite = [d for d in dl
+                            if (np.isfinite(d.intervalx).all() and
+                                (d not in finite_dl))]
+                y_finite = [d for d in dl
+                            if (np.isfinite(d.intervaly).all() and
+                                (d not in finite_dl))]
 
                 dl = finite_dl
                 dl.extend(x_finite)
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -4901,6 +4901,7 @@ def test_minorticks_on():
             ax.minorticks_on()
 
 
+@cleanup
 def test_twinx_knows_limits():
     fig, ax = plt.subplots()
 
