TST: Fix compatibility with Python 2.

QuLogic · QuLogic · commit ebff832c6087 · 2016-10-08T00:26:34.000-04:00
diff --git a/lib/matplotlib/testing/determinism.py b/lib/matplotlib/testing/determinism.py
@@ -2,6 +2,8 @@
 Provides utilities to test output reproducibility.
 """
 
+import six
+
 import io
 import os
 import re
@@ -53,7 +55,13 @@ def _determinism_save(objects='mhi', format="pdf"):
     x = range(5)
     fig.add_subplot(1, 6, 6).plot(x, x)
 
-    fig.savefig(sys.stdout.buffer, format=format)
+    if six.PY2 and format == 'ps':
+        stdout = io.StringIO()
+    else:
+        stdout = getattr(sys.stdout, 'buffer', sys.stdout)
+    fig.savefig(stdout, format=format)
+    if six.PY2 and format == 'ps':
+        sys.stdout.write(stdout.getvalue())
 
     # Restores SOURCE_DATE_EPOCH
     if sde is None:
