remove juggle_3d and update to numpydoc

Raab70 · Raab70 · commit 635e5dc7dbc1 · 2018-06-11T10:38:04.000-05:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -155,20 +155,32 @@ def set_data_3d(self, *args):
         """
         Set the x, y and z data
 
-        ACCEPTS: 3D array (rows are x, y, z) or three 1D arrays
+        Parameters
+        ----------
+        x : array_like
+            The x-data to be plotted
+        y : array_like
+            The y-data to be plotted
+        z : array_like
+            The z-data to be plotted
+
+        Notes
+        -----
+        Accepts x, y, z arguments or a single array_like (x, y, z)
         """
         if len(args) == 1:
-            x, y, z = args[0]
+            self._verts3d = args[0]
         else:
-            x, y, z = args
-
-        self.set_xdata(x)
-        self.set_ydata(y)
-        self.set_3d_properties(zs=z)
+            self._verts3d = args
 
     def get_data_3d(self):
         """
-        Return the xdata, ydata, zdata.
+        Get the current data
+
+        Returns
+        -------
+        verts3d : length-3 tuple or array_likes
+            The current data as a tuple or array_likes
         """
         return self._verts3d
 
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -765,6 +765,18 @@ def test_calling_conventions(self):
             ax.voxels(filled=filled, x=x, y=y, z=z)
 
 
+def test_line3d_set_get_data_3d():
+    x, y, z = [0, 1], [2, 3], [4, 5]
+    x2, y2, z2 = [6, 7], [8, 9], [10, 11]
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
+    lines = ax.plot(x, y, z)
+    line = lines[0]
+    np.testing.assert_array_equal((x, y, z), line.get_data_3d())
+    line.set_data_3d(x2, y2, z2)
+    np.testing.assert_array_equal((x2, y2, z2), line.get_data_3d())
+
+
 def test_inverted_cla():
     # Github PR #5450. Setting autoscale should reset
     # axes to be non-inverted.
