Use context manager in mathtext examples.

QuLogic · QuLogic · commit 83e9ef54ccee · 2021-07-28T16:13:52.000-04:00
diff --git a/examples/text_labels_and_annotations/mathtext_asarray.py b/examples/text_labels_and_annotations/mathtext_asarray.py
@@ -21,10 +21,11 @@ def text_to_rgba(s, *, dpi, **kwargs):
     # (If desired, one can also directly save the image to the filesystem.)
     fig = Figure(facecolor="none")
     fig.text(0, 0, s, **kwargs)
-    buf = BytesIO()
-    fig.savefig(buf, dpi=dpi, format="png", bbox_inches="tight", pad_inches=0)
-    buf.seek(0)
-    rgba = plt.imread(buf)
+    with BytesIO() as buf:
+        fig.savefig(buf, dpi=dpi, format="png", bbox_inches="tight",
+                    pad_inches=0)
+        buf.seek(0)
+        rgba = plt.imread(buf)
     return rgba
 
 
diff --git a/examples/text_labels_and_annotations/mathtext_examples.py b/examples/text_labels_and_annotations/mathtext_examples.py
@@ -107,20 +107,19 @@ def doall():
 if '--latex' in sys.argv:
     # Run: python mathtext_examples.py --latex
     # Need amsmath and amssymb packages.
-    fd = open("mathtext_examples.ltx", "w")
-    fd.write("\\documentclass{article}\n")
-    fd.write("\\usepackage{amsmath, amssymb}\n")
-    fd.write("\\begin{document}\n")
-    fd.write("\\begin{enumerate}\n")
-
-    for i in range(n_lines):
-        s = mathext_demos[i]
-        s = re.sub(r"(?<!\\)\$", "$$", s)
-        fd.write("\\item %s\n" % s)
-
-    fd.write("\\end{enumerate}\n")
-    fd.write("\\end{document}\n")
-    fd.close()
+    with open("mathtext_examples.ltx", "w") as fd:
+        fd.write("\\documentclass{article}\n")
+        fd.write("\\usepackage{amsmath, amssymb}\n")
+        fd.write("\\begin{document}\n")
+        fd.write("\\begin{enumerate}\n")
+
+        for i in range(n_lines):
+            s = mathext_demos[i]
+            s = re.sub(r"(?<!\\)\$", "$$", s)
+            fd.write("\\item %s\n" % s)
+
+        fd.write("\\end{enumerate}\n")
+        fd.write("\\end{document}\n")
 
     subprocess.call(["pdflatex", "mathtext_examples.ltx"])
 else:
