fixup! squash! Fix #72: rewrite the sanitizer to be a treewalker filter only.

Author: gsnedders

html5lib/tests/test_sanitizer.py

@@ -1,7 +1,6 @@
 from __future__ import absolute_import, division, unicode_literals
 
-from html5lib import constants
-from html5lib import parseFragment, serialize
+from html5lib import constants, parseFragment, serialize
 from html5lib.filters import sanitizer
 
 
@@ -29,11 +28,15 @@ def sanitize_html(stream):
 
 
 def test_should_handle_astral_plane_characters():
-    assert '<p>\U0001d4b5 \U0001d538</p>' == sanitize_html("<p>&#x1d4b5; &#x1d538;</p>")
+    sanitized = sanitize_html("<p>&#x1d4b5; &#x1d538;</p>")
+    expected = '<p>\U0001d4b5 \U0001d538</p>'
+    assert expected == sanitized
 
 
 def test_should_allow_relative_uris():
-    assert '<p><a href="/example.com"></a></p>' == sanitize_html('<p><a href="/example.com"></a></p>')
+    sanitized = sanitize_html('<p><a href="/example.com"></a></p>')
+    expected = '<p><a href="/example.com"></a></p>'
+    assert expected == sanitized
 
 
 def test_sanitizer():