Fix saving PNGs to file objects in some places.

QuLogic · QuLogic · commit 566a7af54b1c · 2020-03-08T00:51:45.000-05:00
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -666,7 +666,7 @@ def write_png(self, fname):
         from matplotlib import _png
         im = self.to_rgba(self._A[::-1] if self.origin == 'lower' else self._A,
                           bytes=True, norm=True)
-        with open(fname, "wb") as file:
+        with cbook.open_file_cm(fname, "wb") as file:
             _png.write_png(im, file)
 
     def set_data(self, A):
diff --git a/lib/matplotlib/mathtext.py b/lib/matplotlib/mathtext.py
@@ -3433,7 +3433,7 @@ def to_png(self, filename, texstr, color='black', dpi=120, fontsize=14):
         from matplotlib import _png
         rgba, depth = self.to_rgba(
             texstr, color=color, dpi=dpi, fontsize=fontsize)
-        with open(filename, "wb") as file:
+        with cbook.open_file_cm(filename, "wb") as file:
             _png.write_png(rgba, file)
         return depth
 
