Different way of importing htmlentitydefs for python 3.0 and disabling diff output in test-markdown.py for

the same.

1 files changed, 4 insertions, 2 deletions

diff --git a/test-markdown.py b/test-markdown.py
index 3d9ef66..d2548a0 100755
--- a/test-markdown.py
+++ b/test-markdown.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-import os, difflib, time, gc, codecs, platform
+import os, difflib, time, gc, codecs, platform, sys
 from pprint import pprint
 import textwrap
         
@@ -306,7 +306,9 @@ def testDirectory(dir, measure_time=False, safe_mode=False) :
 
     htmlDiffFile.write("</table>")
 
-    htmlDiffFile.write(diffsBuffer.decode("utf8"))
+    if sys.version < "3.0":
+        htmlDiffFile.write(diffsBuffer.decode("utf8"))
+
     htmlDiffFile.write(FOOTER)
     htmlDiffFile.close()
     print "Diff written to %s" % htmlDiffFilePath