1414from .support import get_data_files ,TestData ,convertExpected
1515
1616from html5lib import html5parser ,treewalkers ,treebuilders ,treeadapters ,constants
17+ from html5lib .filters .lint import Filter as Lint
1718
1819
1920treeTypes = {
@@ -91,7 +92,7 @@ def test_all_tokens(self):
9192p = html5parser .HTMLParser (tree = treeCls ["builder" ])
9293document = p .parse ("<html><head></head><body>a<div>b</div>c</body></html>" )
9394document = treeCls .get ("adapter" ,lambda x :x )(document )
94- output = treeCls ["walker" ](document )
95+ output = Lint ( treeCls ["walker" ](document ) )
9596for expectedToken ,outputToken in zip (expected ,output ):
9697self .assertEqual (expectedToken ,outputToken )
9798
@@ -111,7 +112,7 @@ def runTreewalkerTest(innerHTML, input, expected, errors, treeClass):
111112
112113document = treeClass .get ("adapter" ,lambda x :x )(document )
113114try :
114- output = treewalkers .pprint (treeClass ["walker" ](document ))
115+ output = treewalkers .pprint (Lint ( treeClass ["walker" ](document ) ))
115116output = attrlist .sub (sortattrs ,output )
116117expected = attrlist .sub (sortattrs ,convertExpected (expected ))
117118diff = "" .join (unified_diff ([line + "\n " for line in expected .splitlines ()],