diff options
Diffstat (limited to 'tests/utils.py')
| -rw-r--r-- | tests/utils.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tests/utils.py b/tests/utils.py index 9eb46bf..92d7a4b 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -6,8 +6,8 @@ import codecs import difflib import os import unittest -from StringIO import StringIO +from sqlparse import compat import sqlparse.utils NL = '\n' @@ -31,16 +31,16 @@ class TestCaseBase(unittest.TestCase): def ndiffAssertEqual(self, first, second): """Like failUnlessEqual except use ndiff for readable output.""" if first != second: - sfirst = unicode(first) - ssecond = unicode(second) + sfirst = compat.text_type(first) + ssecond = compat.text_type(second) # Using the built-in .splitlines() method here will cause incorrect # results when splitting statements that have quoted CR/CR+LF # characters. sfirst = sqlparse.utils.split_unquoted_newlines(sfirst) ssecond = sqlparse.utils.split_unquoted_newlines(ssecond) diff = difflib.ndiff(sfirst, ssecond) - fp = StringIO() + fp = compat.StringIO() fp.write(NL) fp.write(NL.join(diff)) - print fp.getvalue() - raise self.failureException, fp.getvalue() + print(fp.getvalue()) + raise self.failureException(fp.getvalue()) |