diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/fdx_test.py | 2 | ||||
-rw-r--r-- | tests/fountain_test.py | 2 | ||||
-rw-r--r-- | tests/richstring_test.py | 3 |
3 files changed, 4 insertions, 3 deletions
diff --git a/tests/fdx_test.py b/tests/fdx_test.py index b6e2bc9..497f908 100644 --- a/tests/fdx_test.py +++ b/tests/fdx_test.py @@ -3,7 +3,7 @@ # http://www.opensource.org/licenses/mit-license.php from testcompat import TestCase -from StringIO import StringIO +from six import StringIO from screenplain.export.fdx import write_text from screenplain.richstring import plain, bold, italic diff --git a/tests/fountain_test.py b/tests/fountain_test.py index e265da9..f100a5c 100644 --- a/tests/fountain_test.py +++ b/tests/fountain_test.py @@ -9,7 +9,7 @@ from screenplain.types import ( Slug, Action, Dialog, DualDialog, Transition, Section, PageBreak ) from screenplain.richstring import plain, italic, empty_string -from StringIO import StringIO +from six import StringIO def parse(lines): diff --git a/tests/richstring_test.py b/tests/richstring_test.py index f2bed5e..ee42087 100644 --- a/tests/richstring_test.py +++ b/tests/richstring_test.py @@ -3,6 +3,7 @@ # http://www.opensource.org/licenses/mit-license.php from testcompat import TestCase +import six from screenplain.richstring import ( RichString, Segment, Bold, Italic, @@ -38,7 +39,7 @@ class RichStringOperatorTests(TestCase): s = bold('Hello') + plain(' there ') + bold('folks') self.assertEquals( u'Hello there folks', - unicode(s) + six.text_type(s) ) def test_eq(self): |