add an image comparison unit test for 3d errorbars

Vlas Sokolov · Vlas Sokolov · commit 45cbc81078b2 · 2020-04-24T10:50:37.000+02:00
diff --git a/examples/mplot3d/errorbar3d.py b/examples/mplot3d/errorbar3d.py
@@ -16,9 +16,6 @@
 t = np.arange(0, 2*np.pi+.1, 0.01)
 x, y, z = np.sin(t), np.cos(3*t), np.sin(5*t)
 
-fig_zerr_kwargs = dict(zerr=0.2, capsize=2)
-fig_xerr_kwargs = dict(xerr=0.2, errorevery=(2, 2))
-
 estep = 15
 zuplims = [True if (not i % estep and i // estep % 3 == 0)
            else False for i in range(t.size)]
diff --git a/lib/mpl_toolkits/tests/baseline_images/test_mplot3d/errorbar3d.png b/lib/mpl_toolkits/tests/baseline_images/test_mplot3d/errorbar3d.png
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -916,3 +916,21 @@ def test_minor_ticks():
     ax.set_yticklabels(["third"], minor=True)
     ax.set_zticks([0.50], minor=True)
     ax.set_zticklabels(["half"], minor=True)
+
+
+@mpl3d_image_comparison(['errorbar3d.png'], tol=0.03)
+def test_errorbar3d():
+    t = np.arange(0, 2*np.pi+.1, 0.01)
+    x, y, z = np.sin(t), np.cos(3*t), np.sin(5*t)
+
+    fig = plt.figure()
+    ax = fig.gca(projection='3d')
+
+    estep = 15
+    zuplims = [True if (not i % estep and i // estep % 3 == 0)
+               else False for i in range(t.size)]
+    zlolims = [True if (not i % estep and i // estep % 3 == 2)
+               else False for i in range(t.size)]
+
+    ax.errorbar(x, y, z, 0.2, zuplims=zuplims, zlolims=zlolims,
+                errorevery=estep)
