Improve stem orientation tests and what's new doc.

QuLogic · QuLogic · commit 13d962aeca92 · 2020-08-14T01:15:25.000-04:00
diff --git a/doc/users/next_whats_new/stem_orientation.rst b/doc/users/next_whats_new/stem_orientation.rst
@@ -1,11 +1,13 @@
 Added *orientation* parameter for stem plots
 --------------------------------------------
-When creating stem plots, you can now pass in an *orientation* argument to
-`.Axes.stem` or `.pyplot.stem`.
 
-Currently, only ``'vertical'`` and ``'horizontal'`` orientations are supported,
-with ``'vertical'`` being the default.
+By default, stem lines are vertical. They can be changed to horizontal using
+the *orientation* parameter of `.Axes.stem` or `.pyplot.stem`:
 
-Example
-```````
-stem(x, x, orientation='horizontal')
+.. plot::
+
+    locs = np.linspace(0.1, 2 * np.pi, 25)
+    heads = np.cos(locs)
+
+    fig, ax = plt.subplots()
+    ax.stem(locs, heads, orientation='horizontal')
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -3245,14 +3245,13 @@ def test_hist_stacked_weighted():
 @image_comparison(['stem.png'], style='mpl20', remove_text=True)
 def test_stem(use_line_collection):
     x = np.linspace(0.1, 2 * np.pi, 100)
-    args = (x, np.cos(x))
-    # Label is a single space to force a legend to be drawn, but to avoid any
-    # text being drawn
-    kwargs = dict(linefmt='C2-.', markerfmt='k+', basefmt='C1-.',
-                  label=' ', use_line_collection=use_line_collection)
 
     fig, ax = plt.subplots()
-    ax.stem(*args, **kwargs)
+    # Label is a single space to force a legend to be drawn, but to avoid any
+    # text being drawn
+    ax.stem(x, np.cos(x),
+            linefmt='C2-.', markerfmt='k+', basefmt='C1-.', label=' ',
+            use_line_collection=use_line_collection)
 
     ax.legend()
 
@@ -3284,12 +3283,11 @@ def test_stem_dates():
 @image_comparison(['stem_orientation.png'], style='mpl20', remove_text=True)
 def test_stem_orientation(use_line_collection):
     x = np.linspace(0.1, 2*np.pi, 50)
-    args = (x, np.cos(x))
-    kwargs = dict(linefmt='C2-.', markerfmt='kx', basefmt='C1-.',
-                  use_line_collection=use_line_collection)
 
     fig, ax = plt.subplots()
-    ax.stem(*args, **kwargs, orientation='horizontal')
+    ax.stem(x, np.cos(x),
+            linefmt='C2-.', markerfmt='kx', basefmt='C1-.',
+            use_line_collection=use_line_collection, orientation='horizontal')
 
 
 @image_comparison(['hist_stacked_stepfilled_alpha'])
