Add unit test to ensure that rendering complexity exceeded exception is properly propagated to Python.

mdboom · mdboom · commit 6e629b597d68 · 2010-06-23T14:44:32.000Z
svn path=/trunk/matplotlib/; revision=8459
diff --git a/lib/matplotlib/tests/test_simplification.py b/lib/matplotlib/tests/test_simplification.py
@@ -6,6 +6,10 @@
 from pylab import *
 import numpy as np
 from matplotlib import patches, path, transforms
+
+from nose.tools import raises
+import cStringIO
+
 nan = np.nan
 Path = path.Path
 
@@ -165,6 +169,24 @@ def test_start_with_moveto():
     assert len(segs) == 1
     assert segs[0][1] == Path.MOVETO
 
+@raises(OverflowError)
+def test_throw_rendering_complexity_exceeded():
+    rcParams['path.simplify'] = False
+
+    xx = np.arange(200000)
+    yy = np.random.rand(200000)
+    yy[1000] = np.nan
+    fig = plt.figure()
+    ax = fig.add_subplot(111)
+    ax.plot(xx, yy)
+    try:
+        fig.savefig(cStringIO.StringIO())
+    except e:
+        raise e
+    else:
+        rcParams['path.simplify'] = True
+
+
 if __name__=='__main__':
     import nose
     nose.runmodule(argv=['-s','--with-doctest'], exit=False)
