Add a shape assert

apaszke · apaszke · commit ed055567c7a5 · 2020-03-11T12:49:45.000+01:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -577,7 +577,10 @@ def set_zsort(self, zsort):
 
     def get_vector(self, segments3d):
         """Optimize points for projection."""
+        # segments3d.shape == (num_faces, num_vertices, 3)
         if isinstance(segments3d, np.ndarray):
+            assert segments3d.dim == 3 and segments3d.shape[-1] == 3, \
+                "Incorrect shape of vertices of Poly3DCollection"
             self._segments = segments3d
         else:
             num_faces = len(segments3d)
