TST: Check just one startxref and file size not ballooning

jklymak · jklymak · commit 621393b00c70 · 2017-11-10T11:00:37.000-08:00
diff --git a/lib/matplotlib/tests/test_backend_pdf.py b/lib/matplotlib/tests/test_backend_pdf.py
@@ -7,6 +7,7 @@
 
 import io
 import os
+import sys
 import tempfile
 
 import pytest
@@ -71,6 +72,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 sys.getsizeof(pdfio) < 40000
+
+
 def test_multipage_keep_empty():
     from matplotlib.backends.backend_pdf import PdfPages
     from tempfile import NamedTemporaryFile
