1 files changed, 8 insertions, 6 deletions
diff --git a/tests/test_py_domain.py b/tests/test_py_domain.py
index 68634d83..87f6eb98 100644
--- a/tests/test_py_domain.py
+++ b/tests/test_py_domain.py
@@ -9,6 +9,8 @@
     :license: BSD, see LICENSE for details.
 """
 
+from six import text_type
+
 from sphinx import addnodes
 from sphinx.domains.python import py_sig_re, _pseudo_parse_arglist
 
@@ -26,19 +28,19 @@ def parse(sig):
 def test_function_signatures():
 
     rv = parse('func(a=1) -> int object')
-    assert unicode(rv) == u'a=1'
+    assert text_type(rv) == u'a=1'
 
     rv = parse('func(a=1, [b=None])')
-    assert unicode(rv) == u'a=1, [b=None]'
+    assert text_type(rv) == u'a=1, [b=None]'
 
     rv = parse('func(a=1[, b=None])')
-    assert unicode(rv) == u'a=1, [b=None]'
+    assert text_type(rv) == u'a=1, [b=None]'
 
     rv = parse("compile(source : string, filename, symbol='file')")
-    assert unicode(rv) == u"source : string, filename, symbol='file'"
+    assert text_type(rv) == u"source : string, filename, symbol='file'"
 
     rv = parse('func(a=[], [b=None])')
-    assert unicode(rv) == u'a=[], [b=None]'
+    assert text_type(rv) == u'a=[], [b=None]'
 
     rv = parse('func(a=[][, b=None])')
-    assert unicode(rv) == u'a=[], [b=None]'
+    assert text_type(rv) == u'a=[], [b=None]'