diff options
Diffstat (limited to 'src/beautifulsoup/tests/helpers.py')
-rw-r--r-- | src/beautifulsoup/tests/helpers.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/src/beautifulsoup/tests/helpers.py b/src/beautifulsoup/tests/helpers.py index 219d95c..d237556 100644 --- a/src/beautifulsoup/tests/helpers.py +++ b/src/beautifulsoup/tests/helpers.py @@ -3,15 +3,21 @@ import unittest from beautifulsoup import BeautifulSoup from beautifulsoup.element import SoupStrainer +from beautifulsoup.builder.lxml_builder import LXMLTreeBuilder class SoupTest(unittest.TestCase): default_builder = None def assertSoupEquals(self, to_parse, compare_parsed_to=None): - obj = BeautifulSoup(to_parse, builder=self.default_builder) + builder = self.default_builder + if builder is None: + builder = LXMLTreeBuilder() + obj = BeautifulSoup(to_parse, builder=builder) if compare_parsed_to is None: compare_parsed_to = to_parse - self.assertEquals(obj.decode(), compare_parsed_to) + self.assertEquals( + obj.decode(), + builder.test_fragment_to_document(compare_parsed_to)) |