Fix the unitary tests.

1 files changed, 15 insertions, 6 deletions

diff --git a/creole/tests/test_creole2html.py b/creole/tests/test_creole2html.py
index 9915dc1..45e73dd 100644
--- a/creole/tests/test_creole2html.py
+++ b/creole/tests/test_creole2html.py
@@ -28,6 +28,7 @@ except ImportError:
 
 from creole.tests.utils.base_unittest import BaseCreoleTest
 from creole.tests import test_macros
+from creole.py3compat import PY3
 
 from creole import creole2html
 from creole.shared import example_macros
@@ -57,12 +58,20 @@ class TestCreole2html(unittest.TestCase):
         error_msg = my_stderr.getvalue()
 
         # Check if we get a traceback information into our stderr handler
-        must_have = (
-            "Traceback",
-            "KeyError:",
-            "KeyError: 'notexist1'",
-            "KeyError: 'notexist2'",
-        )
+        if PY3:
+            must_have = (
+                "Traceback",
+                "KeyError:",
+                "KeyError: 'notexist1'",
+                "KeyError: 'notexist2'",
+            )
+        else:
+            must_have = (
+                "Traceback",
+                "KeyError:",
+                "KeyError: u'notexist1'",
+                "KeyError: u'notexist2'",
+            )
 
         for part in must_have:
             self.assertIn(part, error_msg)