summaryrefslogtreecommitdiff
path: root/bs4/tests/test_builder_registry.py
diff options
context:
space:
mode:
Diffstat (limited to 'bs4/tests/test_builder_registry.py')
-rw-r--r--bs4/tests/test_builder_registry.py17
1 files changed, 9 insertions, 8 deletions
diff --git a/bs4/tests/test_builder_registry.py b/bs4/tests/test_builder_registry.py
index 5fa874c..9327174 100644
--- a/bs4/tests/test_builder_registry.py
+++ b/bs4/tests/test_builder_registry.py
@@ -10,22 +10,23 @@ from bs4.builder import (
TreeBuilderRegistry,
)
-try:
+from . import (
+ HTML5LIB_PRESENT,
+ LXML_PRESENT,
+)
+
+if HTML5LIB_PRESENT:
from bs4.builder import HTML5TreeBuilder
- HTML5LIB_PRESENT = True
-except ImportError:
- HTML5LIB_PRESENT = False
-try:
+if LXML_PRESENT:
from bs4.builder import (
LXMLTreeBuilderForXML,
LXMLTreeBuilder,
)
- LXML_PRESENT = True
-except ImportError:
- LXML_PRESENT = False
+# TODO: Split out the lxml and html5lib tests into their own classes
+# and gate with pytest.mark.skipIf.
class TestBuiltInRegistry(object):
"""Test the built-in registry with the default builders registered."""