Bugfix

scottshambaugh · scottshambaugh · commit 1c715f930477 · 2024-10-16T16:23:18.000Z
Tests

Tests
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -455,7 +455,7 @@ def do_3d_projection(self):
             all_points = np.ma.vstack(segments)
             masked_points = np.ma.column_stack([*_viewlim_mask(*all_points.T,
                                                                self.axes)])
-            segment_lengths = [segment.shape[0] for segment in segments]
+            segment_lengths = [np.asanyarray(segment).shape[0] for segment in segments]
             segments = np.split(masked_points, np.cumsum(segment_lengths[:-1]))
         xyslist = [proj3d._proj_trans_points(points, self.axes.M)
                    for points in segments]
diff --git a/lib/mpl_toolkits/mplot3d/tests/test_axes3d.py b/lib/mpl_toolkits/mplot3d/tests/test_axes3d.py
@@ -1389,14 +1389,15 @@ def test_axlim_clip(fig_test, fig_ref):
     Z = X + Y
     ax.plot_surface(X, Y, Z, facecolor='C1', edgecolors=None,
                     rcount=50, ccount=50, axlim_clip=True)
+    ax.plot_wireframe(X, Y, Z + 0.5, color='C2', rcount=50, ccount=50, axlim_clip=True)
     # This ax.plot is to cover the extra surface edge which is not clipped out
     ax.plot([0.5, 0.5], [0, 1], [0.5, 1.5],
             color='k', linewidth=3, zorder=5, axlim_clip=True)
     ax.scatter(X.ravel(), Y.ravel(), Z.ravel() + 1, axlim_clip=True)
     ax.quiver(X.ravel(), Y.ravel(), Z.ravel() + 2,
               0*X.ravel(), 0*Y.ravel(), 0*Z.ravel() + 1,
               arrow_length_ratio=0, axlim_clip=True)
-    ax.plot(X[0], Y[0], Z[0] + 3, color='C2', axlim_clip=True)
+    ax.plot(X[0], Y[0], Z[0] + 3, color='C3', axlim_clip=True)
     ax.text(1.1, 0.5, 4, 'test', axlim_clip=True)  # won't be visible
     ax.set(xlim=(0, 0.5), ylim=(0, 1), zlim=(0, 5))
 
@@ -1408,13 +1409,15 @@ def test_axlim_clip(fig_test, fig_ref):
     Z = Z[idx].reshape(11, 6)
     ax.plot_surface(X, Y, Z, facecolor='C1', edgecolors=None,
                     rcount=50, ccount=50, axlim_clip=False)
+    ax.plot_wireframe(X, Y, Z + 0.5, color='C2',
+                      rcount=50, ccount=50, axlim_clip=False)
     ax.plot([0.5, 0.5], [0, 1], [0.5, 1.5],
             color='k', linewidth=3, zorder=5, axlim_clip=False)
     ax.scatter(X.ravel(), Y.ravel(), Z.ravel() + 1, axlim_clip=False)
     ax.quiver(X.ravel(), Y.ravel(), Z.ravel() + 2,
               0*X.ravel(), 0*Y.ravel(), 0*Z.ravel() + 1,
               arrow_length_ratio=0, axlim_clip=False)
-    ax.plot(X[0], Y[0], Z[0] + 3, color='C2', axlim_clip=False)
+    ax.plot(X[0], Y[0], Z[0] + 3, color='C3', axlim_clip=False)
     ax.set(xlim=(0, 0.5), ylim=(0, 1), zlim=(0, 5))
 
 
