Adding test test_relim_visible_only

fariza · fariza · commit 94bdfd623aae · 2013-09-16T13:54:04.000-04:00
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -1737,6 +1737,32 @@ def test_empty_shared_subplots():
     assert y1 >= 6
 
 
+@cleanup
+def test_relim_visible_only():
+    x1 = (0., 10.)
+    y1 = (0., 10.)
+    x2 = (-10., 20.)
+    y2 = (-10., 30.)
+
+    fig = matplotlib.figure.Figure()
+    ax = fig.add_subplot(111)
+    ax.plot(x1, y1)
+    assert ax.get_xlim() == x1
+    assert ax.get_ylim() == y1
+    l = ax.plot(x2, y2)
+    assert ax.get_xlim() == x2
+    assert ax.get_ylim() == y2
+    l[0].set_visible(False)
+    assert ax.get_xlim() == x2
+    assert ax.get_ylim() == y2
+    
+    ax.relim(visible_only=True)
+    ax.autoscale_view()
+
+    assert ax.get_xlim() == x1
+    assert ax.get_ylim() == y1
+
+
 if __name__ == '__main__':
     import nose
     nose.runmodule(argv=['-s', '--with-doctest'], exit=False)
