Patch doctests to work with Python 3python3

2 files changed, 25 insertions, 2 deletions

diff --git a/bin/test b/bin/test
index dcefe4f..da49a7e 100755
--- a/bin/test
+++ b/bin/test
@@ -1,3 +1,2 @@
 #!/bin/bash
-nosetests --nocapture --with-doctest --doctest-tests $* && \
-  pep8 screenplain tests
+python test.py && pep8 screenplain tests
diff --git a/test.py b/test.py
new file mode 100644
index 0000000..c4e5887
--- /dev/null
+++ b/test.py
@@ -0,0 +1,24 @@
+"""Runs nosetest after preparing the test cases.
+
+Removes the leading u from unicode literals so
+Python 3 doctests won't fail.
+
+"""
+
+from screenplain import richstring
+import re
+
+unicode_literal = re.compile(r'u([\'"])')
+
+for n in (
+    'parse_emphasis',
+    '_unescape',
+    '_demagic_literals',
+):
+    attr = getattr(richstring, n)
+    old_doc = getattr(attr, '__doc__')
+    new_doc = unicode_literal.sub(r'\1', old_doc)
+    setattr(attr, '__doc__', new_doc)
+
+import nose
+nose.main(argv='nosetests --nocapture --with-doctest --doctest-tests'.split())