Update test_lines.py

ImportanceOfBeingErnest · web-flow · commit e3d4e5fa7348 · 2018-01-28T23:13:51.000+01:00
Add a test for setting drawstyle after line creation
diff --git a/lib/matplotlib/tests/test_lines.py b/lib/matplotlib/tests/test_lines.py
@@ -129,6 +129,18 @@ def test_valid_drawstyles():
     with pytest.raises(ValueError):
         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():
