Add basic test for switching autoscale status.

anntzer · anntzer · commit 6df9101a44f2 · 2015-11-22T18:02:44.000-08:00
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -4105,6 +4105,24 @@ def test_violin_point_mass():
     """Violin plot should handle point mass pdf gracefully."""
     plt.violinplot(np.array([0, 0]))
 
+
+@cleanup
+def test_autoscale():
+    plt.plot([0, 1], [0, 1])
+    assert_equal(plt.xlim(), (0, 1))
+    plt.autoscale(False)
+    plt.plot([2, 3], [2, 3])
+    assert_equal(plt.xlim(), (0, 1))
+    plt.autoscale(True)
+    plt.plot([1, 2], [1, 2])
+    assert_equal(plt.xlim(), (0, 3))
+    with plt.autoscale(False):
+        plt.plot([3, 4], [3, 4])
+    assert_equal(plt.xlim(), (0, 3))
+    plt.plot([4, 5], [4, 5])
+    assert_equal(plt.xlim(), (0, 5))
+
+
 if __name__ == '__main__':
     import nose
     import sys
