fixup! Add a test for HTMLParser(debug=True)

gsnedders · gsnedders · commit 76bf2420d202 · 2016-05-22T02:41:13.000+01:00
diff --git a/html5lib/tests/test_parser2.py b/html5lib/tests/test_parser2.py
@@ -1,6 +1,6 @@
 from __future__ import absolute_import, division, unicode_literals
 
-from six import PY2
+from six import PY2, text_type
 
 import io
 
@@ -84,7 +84,7 @@ def test_debug_log():
 
     if PY2:
         for i, log in enumerate(expected):
-            log = [x.encode("ascii") if isinstance(x, unicode) else x for x in log]
+            log = [x.encode("ascii") if isinstance(x, text_type) else x for x in log]
             expected[i] = tuple(log)
 
     assert parser.log == expected
