diff options
Diffstat (limited to 'docutils/test')
| -rwxr-xr-x | docutils/test/test_utils.py | 7 | ||||
| -rwxr-xr-x | docutils/test/test_writers/test_docutils_xml.py | 7 |
2 files changed, 12 insertions, 2 deletions
diff --git a/docutils/test/test_utils.py b/docutils/test/test_utils.py index e5036bc7b..0f53355eb 100755 --- a/docutils/test/test_utils.py +++ b/docutils/test/test_utils.py @@ -9,12 +9,17 @@ Test module for utils/__init__.py. """ -from io import StringIO import os +import sys import unittest from DocutilsTestSupport import docutils, utils, nodes +if sys.version_info >= (3, 0): + from io import StringIO +else: + from StringIO import StringIO + class ReporterTests(unittest.TestCase): diff --git a/docutils/test/test_writers/test_docutils_xml.py b/docutils/test/test_writers/test_docutils_xml.py index c39523dae..abab5742a 100755 --- a/docutils/test/test_writers/test_docutils_xml.py +++ b/docutils/test/test_writers/test_docutils_xml.py @@ -15,12 +15,17 @@ Test for docutils XML writer. """ from __future__ import absolute_import -from StringIO import StringIO +import sys from . import DocutilsTestSupport # must be imported before docutils import docutils import docutils.core +if sys.version_info >= (3, 0): + from io import StringIO +else: + from StringIO import StringIO + # sample strings # -------------- |
