Forward keyword arguments onto Poly3DCollection

eric-wieser · eric-wieser · commit 1e73931e7633 · 2017-07-14T15:36:11.000+01:00
Also test that the result is indexable
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -2745,7 +2745,7 @@ def calc_arrow(uvw, angle=15):
 
     quiver3D = quiver
 
-    def voxels(self, filled, color=None):
+    def voxels(self, filled, color=None, **kwargs):
         """
         Plot a set of filled voxels
 
@@ -2766,6 +2766,9 @@ def voxels(self, filled, color=None):
             indicating what color to draw the faces of the voxels. If None,
             plot all voxels in the same color, the next in the color sequence.
 
+        Any additional keyword arguments are passed onto
+        :func:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection`
+
         Returns
         -------
         faces : dict
@@ -2863,7 +2866,10 @@ def permutation_matrices(n):
         # iterate over the faces, and generate a Poly3DCollection for each voxel
         polygons = {}
         for coord, faces in voxel_faces.items():
-            poly = art3d.Poly3DCollection(faces, facecolors=color[coord])
+            poly = art3d.Poly3DCollection(faces,
+                facecolors=color[coord],
+                **kwargs
+            )
             self.add_collection3d(poly)
             polygons[coord] = poly
 
diff --git a/lib/mpl_toolkits/tests/baseline_images/test_mplot3d/voxels-edge-style.png b/lib/mpl_toolkits/tests/baseline_images/test_mplot3d/voxels-edge-style.png
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -582,6 +582,22 @@ def test_simple(self):
         voxels = (x == y) | (y == z)
         ax.voxels(voxels)
 
+    @image_comparison(
+        baseline_images=['voxels-edge-style'],
+        extensions=['png'],
+        remove_text=True,
+        style='default'
+    )
+    def test_edge_style(self):
+        fig, ax = plt.subplots(subplot_kw={"projection": "3d"})
+
+        x, y, z = np.indices((5, 5, 4))
+        voxels = ((x - 2)**2 + (y - 2)**2 + (z-1.5)**2) < 2.2**2
+        v = ax.voxels(voxels, linewidths=3, edgecolor='C1')
+
+        # change the edge color of one voxel
+        v[max(v.keys())].set_edgecolor('C2')
+
 
     @image_comparison(
         baseline_images=['voxels-named-colors'],
