Add setters for more overridden 3D collection properties.

QuLogic · QuLogic · commit a615aed88fee · 2020-09-17T02:16:15.000-04:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -507,6 +507,14 @@ def set_edgecolor(self, c):
         super().set_edgecolor(c)
         self._edgecolor3d = self.get_edgecolor()
 
+    def set_sizes(self, sizes, dpi=72.0):
+        super().set_sizes(sizes, dpi=dpi)
+        self._sizes3d = self.get_sizes()
+
+    def set_linewidth(self, lw):
+        super().set_linewidth(lw)
+        self._linewidth3d = self.get_linewidth()
+
     def do_3d_projection(self, renderer):
         xs, ys, zs = self._offsets3d
         vxs, vys, vzs, vis = proj3d.proj_transform_clip(xs, ys, zs, renderer.M)
@@ -542,8 +550,8 @@ def do_3d_projection(self, renderer):
 
         super().set_edgecolor(ecs)
         super().set_facecolor(fcs)
-        self.set_sizes(sizes)
-        self.set_linewidth(lws)
+        super().set_sizes(sizes)
+        super().set_linewidth(lws)
 
         PathCollection.set_offsets(self, vps)
 
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -251,10 +251,12 @@ def test_scatter3d_modification(fig_ref, fig_test):
                         marker='o')
     c.set_facecolor('C1')
     c.set_edgecolor('C2')
+    c.set_sizes(np.full(10, 75))
+    c.set_linewidths(3)
 
     ax_ref = fig_ref.add_subplot(projection='3d')
     ax_ref.scatter(np.arange(10), np.arange(10), np.arange(10), marker='o',
-                   facecolor='C1', edgecolor='C2')
+                   facecolor='C1', edgecolor='C2', s=75, linewidths=3)
 
 
 @pytest.mark.parametrize('depthshade', [True, False])
