Remove gca() warning from test_axes

dstansby · dstansby · commit 9c49521ac42e · 2018-03-04T19:07:52.000Z
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -1,6 +1,3 @@
-from __future__ import absolute_import, division, print_function
-
-import six
 from itertools import chain, product
 from distutils.version import LooseVersion
 import io
@@ -602,14 +599,14 @@ def test_const_xy():
                   style='default')
 def test_polar_wrap():
     fig = plt.figure()
-    plt.subplot(111, polar=True)
+    plt.subplot(111, polar=True, label='First axes')
     plt.polar(np.deg2rad([179, -179]), [0.2, 0.1], "b.-")
-    plt.polar(np.deg2rad([179,  181]), [0.2, 0.1], "g.-")
+    plt.polar(np.deg2rad([179, 181]), [0.2, 0.1], "g.-")
     plt.rgrids([0.05, 0.1, 0.15, 0.2, 0.25, 0.3])
     assert len(fig.axes) == 1, 'More than one polar axes created.'
 
     fig = plt.figure()
-    plt.subplot(111, polar=True)
+    plt.subplot(111, polar=True, label='Second axes')
     plt.polar(np.deg2rad([2, -2]), [0.2, 0.1], "b.-")
     plt.polar(np.deg2rad([2, 358]), [0.2, 0.1], "g.-")
     plt.polar(np.deg2rad([358, 2]), [0.2, 0.1], "r.-")
@@ -705,11 +702,10 @@ def test_polar_rlabel_position():
     ax.tick_params(rotation='auto')
 
 
-@image_comparison(baseline_images=['polar_theta_wedge'], style='default',
-                  tol=0.01 if six.PY2 else 0)
+@image_comparison(baseline_images=['polar_theta_wedge'], style='default')
 def test_polar_theta_limits():
     r = np.arange(0, 3.0, 0.01)
-    theta = 2*np.pi*r
+    theta = 2 * np.pi * r
 
     theta_mins = np.arange(15.0, 361.0, 90.0)
     theta_maxs = np.arange(50.0, 361.0, 90.0)
@@ -4329,7 +4325,7 @@ def test_twin_spines():
     def make_patch_spines_invisible(ax):
         ax.set_frame_on(True)
         ax.patch.set_visible(False)
-        for sp in six.itervalues(ax.spines):
+        for sp in ax.spines.values:
             sp.set_visible(False)
 
     fig = plt.figure(figsize=(4, 3))
