Py3 test case fix

diff --git a/tests/run/set.pyx b/tests/run/set.pyx
index 24e853ee549337af0c0059c6fb0bcced2090838a..ab14513774084fea0e890d38ac8cfc3783bfa10d 100644 (file)
--- a/tests/run/set.pyx
+++ b/tests/run/set.pyx
@@ -2,12 +2,12 @@ __doc__ = u"""
 >>> type(test_set_literal()) is _set
 True
 >>> sorted(test_set_literal())
-['a', 'b', 1]
+[u'a', u'b', 1]
 
 >>> type(test_set_add()) is _set
 True
 >>> sorted(test_set_add())
-['a', 1]
+[u'a', 1]
 
 >>> type(test_set_list_comp()) is _set
 True
@@ -27,21 +27,25 @@ True
 >>> type(test_set_discard()) is _set
 True
 >>> sorted(test_set_discard())
-['12', 233]
+[u'12', 233]
 """
 
+import sys
+if sys.version_info[0] >= 3:
+    __doc__ = __doc__.replace(u"u'", u"'").replace(u'u"', u'"')
+
 # Py2.3 doesn't have the 'set' builtin type, but Cython does :)
 _set = set
 
 def test_set_literal():
-    cdef set s1 = {1,'a',1,'b','a'}
+    cdef set s1 = {1,u'a',1,u'b',u'a'}
     return s1
 
 def test_set_add():
     cdef set s1
     s1 = set([1])
     s1.add(1)
-    s1.add('a')
+    s1.add(u'a')
     s1.add(1)
     return s1
 
@@ -59,17 +63,17 @@ def test_set_list_comp():
 def test_set_pop():
     cdef set s1
     s1 = set()
-    s1.add('2')
+    s1.add(u'2')
     two = s1.pop()
     return s1
 
 def test_set_discard():
     cdef set s1
     s1 = set()
-    s1.add('12')
+    s1.add(u'12')
     s1.add(3)
     s1.add(233)
-    s1.discard('3')
+    s1.discard(u'3')
     s1.discard(3)
     return s1