add a _u function to imitate unicode literals on python3

1 files changed, 12 insertions, 0 deletions

diff --git a/testtools/utils.py b/testtools/utils.py
index 2f502d0..0d9c3f4 100644
--- a/testtools/utils.py
+++ b/testtools/utils.py
@@ -2,12 +2,24 @@
 
 """Utilities for dealing with stuff in unittest."""
 
+
+import sys
+
 __metaclass__ = type
 __all__ = [
     'iterate_tests',
     ]
 
 
+if sys.version_info > (3, 0):
+    def _u(s):
+        """Replacement for u'some string' in Python 3."""
+        return s
+else:
+    def _u(s):
+        return unicode(s)
+
+
 def iterate_tests(test_suite_or_case):
     """Iterate through all of the test cases in `test_suite_or_case`."""
     try: