FIX: get tests working

jklymak · jklymak · commit 31b6fe7376a2 · 2021-02-07T09:43:39.000-08:00
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -382,6 +382,7 @@ def test_EllipseCollection():
 
 @image_comparison(['polycollection_close.png'], remove_text=True)
 def test_polycollection_close():
+    from mpl_toolkits.mplot3d import Axes3D
 
     vertsQuad = [
         [[0., 0.], [0., 1.], [1., 1.], [1., 0.]],
@@ -390,7 +391,7 @@ def test_polycollection_close():
         [[3., 0.], [3., 1.], [4., 1.], [4., 0.]]]
 
     fig = plt.figure()
-    ax = fig.add_subplot(projection='3d')
+    ax = fig.add_axes(Axes3D(fig, add=False))
 
     colors = ['r', 'g', 'b', 'y', 'k']
     zpos = list(range(5))
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -3,7 +3,7 @@
 
 import pytest
 
-from mpl_toolkits.mplot3d import axes3d, proj3d, art3d
+from mpl_toolkits.mplot3d import Axes3D, axes3d, proj3d, art3d
 import matplotlib as mpl
 from matplotlib.backend_bases import MouseButton
 from matplotlib import cm
@@ -725,7 +725,8 @@ def test_add_collection3d_zs_scalar():
 @mpl3d_image_comparison(['axes3d_labelpad.png'], remove_text=False)
 def test_axes3d_labelpad():
     fig = plt.figure()
-    ax = fig.add_subplot(projection='3d')
+    ax = Axes3D(fig, add=False)
+    fig.add_axes(ax)
     # labelpad respects rcParams
     assert ax.xaxis.labelpad == mpl.rcParams['axes.labelpad']
     # labelpad can be set in set_label
