77except AttributeError :
88unittest .TestCase .assertEqual = unittest .TestCase .assertEquals
99
10+ import pytest
11+
1012from .support import treeTypes
1113
1214from html5lib import html5parser ,treewalkers
@@ -38,6 +40,8 @@ def test_all_tokens(self):
3840 {'type' :'EndTag' ,'namespace' :'http://www.w3.org/1999/xhtml' ,'name' :'html' }
3941 ]
4042for treeName ,treeCls in sorted (treeTypes .items ()):
43+ if treeCls is None :
44+ pytest .skip ("Treebuilder not loaded" )
4145p = html5parser .HTMLParser (tree = treeCls ["builder" ])
4246document = p .parse ("<html><head></head><body>a<div>b</div>c</body></html>" )
4347document = treeCls .get ("adapter" ,lambda x :x )(document )
@@ -61,6 +65,8 @@ def set_attribute_on_first_child(docfrag, name, value, treeName):
6165def runTreewalkerEditTest (intext ,expected ,attrs_to_add ,tree ):
6266"""tests what happens when we add attributes to the intext"""
6367treeName ,treeClass = tree
68+ if treeClass is None :
69+ pytest .skip ("Treebuilder not loaded" )
6470parser = html5parser .HTMLParser (tree = treeClass ["builder" ])
6571document = parser .parseFragment (intext )
6672for nom ,val in attrs_to_add :