cleaning up code

Osarumwense · Osarumwense · commit bfe56366f6c8 · 2018-03-08T17:54:34.000-05:00
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -1568,10 +1568,7 @@ def plot(self, xs, ys, *args, **kwargs):
         for line in lines:
             art3d.line_2d_to_3d(line, zs=zs, zdir=zdir)
 
-        # when transform from 2d to 3d, dataset changes, update the dataset
-        # for setting axes bounds 
-        (xs, ys, zs) = art3d.juggle_axes(xs, ys, zs, zdir)
-
+        xs, ys, zs = art3d.juggle_axes(xs, ys, zs, zdir)
         self.auto_scale_xyz(xs, ys, zs, had_data)
         return lines
 
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -174,6 +174,7 @@ def test_scatter3d_color():
     ax.scatter(np.arange(10, 20), np.arange(10, 20), np.arange(10, 20),
                color='b', marker='s')
 
+
 @image_comparison(baseline_images=['plot_3d_from_2d'], remove_text=True,
                   extensions=['png'])
 def test_plot_3d_from_2d():
@@ -183,6 +184,7 @@ def test_plot_3d_from_2d():
     ys = np.arange(5, 10)
     ax.plot(xs, ys, zs=0, zdir='x')
 
+
 @image_comparison(baseline_images=['surface3d'], remove_text=True)
 def test_surface3d():
     fig = plt.figure()
