Fix test in Py2.

c02c3f36 · Stefan Behnel · 9bc7db3d · c02c3f36
Commit c02c3f36 authored Jun 04, 2020 by Stefan Behnel
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

tests/build/cythonize_options.srctree tests/build/cythonize_options.srctree +5 -1

No files found.
--- a/tests/build/cythonize_options.srctree
+++ b/tests/build/cythonize_options.srctree
@@ -9,7 +9,11 @@ from Cython.Build.Dependencies import cythonize
 import sys
 from distutils.core import setup
-from io import StringIO
+try:
+    # io.StringIO cannot handle 'str' in Py2
+    from StringIO import StringIO
+except ImportError:
+    from io import StringIO
 old_stderr = sys.stderr
 captured = sys.stderr = StringIO()