Merge pull request #10340 from ImportanceOfBeingErnest/patch-7

anntzer · web-flow · commit 6839a6dde7ce · 2018-01-28T21:16:36.000-08:00
update set_drawstyle
diff --git a/lib/matplotlib/lines.py b/lib/matplotlib/lines.py
@@ -1002,6 +1002,8 @@ def set_drawstyle(self, drawstyle):
             raise ValueError('Unrecognized drawstyle {!r}'.format(drawstyle))
         if self._drawstyle != drawstyle:
             self.stale = True
+            # invalidate to trigger a recache of the path
+            self._invalidx = True
         self._drawstyle = drawstyle
 
     def set_linewidth(self, w):
diff --git a/lib/matplotlib/tests/test_lines.py b/lib/matplotlib/tests/test_lines.py
@@ -130,6 +130,19 @@ def test_valid_drawstyles():
         line.set_drawstyle('foobar')
 
 
+def test_set_drawstyle():
+    x = np.linspace(0, 2*np.pi, 10)
+    y = np.sin(x)
+
+    fig, ax = plt.subplots()
+    line, = ax.plot(x, y)
+    line.set_drawstyle("steps-pre")
+    assert len(line.get_path().vertices) == 2*len(x)-1
+
+    line.set_drawstyle("default")
+    assert len(line.get_path().vertices) == len(x)
+
+
 @image_comparison(baseline_images=['line_collection_dashes'], remove_text=True)
 def test_set_line_coll_dash_image():
     fig = plt.figure()
