summaryrefslogtreecommitdiff
path: root/src/beautifulsoup/tests/helpers.py
blob: 219d95c3cd524c3c2829331ca37178c36e7daa88 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
"""Helper classes for tests."""

import unittest
from beautifulsoup import BeautifulSoup
from beautifulsoup.element import SoupStrainer

class SoupTest(unittest.TestCase):

    default_builder = None

    def assertSoupEquals(self, to_parse, compare_parsed_to=None):
        obj = BeautifulSoup(to_parse, builder=self.default_builder)
        if compare_parsed_to is None:
            compare_parsed_to = to_parse

        self.assertEquals(obj.decode(), compare_parsed_to)