TST: Check just one startxref and file size not ballooning

jklymak · jklymak · commit 02cf299d1e4d · 2017-11-10T10:12:17.000-08:00
diff --git a/lib/matplotlib/tests/test_backend_pdf.py b/lib/matplotlib/tests/test_backend_pdf.py
@@ -71,6 +71,19 @@ def test_multipage_pagecount():
         assert pdf.get_pagecount() == 2
 
 
+def test_multipage_properfinalize():
+    pdfio = io.BytesIO()
+    with PdfPages(pdfio) as pdf:
+        for i in range(10):
+            fig = plt.figure()
+            ax = fig.add_subplot(111)
+            ax.set_title('This is a long title')
+            fig.savefig(pdf, format="pdf")
+    pdfio.seek(0)
+    assert sum(b'startxref' in line for line in pdfio) == 1
+    assert pdfio.getbuffer().nbytes < 25000
+
+
 def test_multipage_keep_empty():
     from matplotlib.backends.backend_pdf import PdfPages
     from tempfile import NamedTemporaryFile
