diff options
Diffstat (limited to 'bs4/tests/test_htmlparser.py')
-rw-r--r-- | bs4/tests/test_htmlparser.py | 25 |
1 files changed, 14 insertions, 11 deletions
diff --git a/bs4/tests/test_htmlparser.py b/bs4/tests/test_htmlparser.py index c6a6691..7be6493 100644 --- a/bs4/tests/test_htmlparser.py +++ b/bs4/tests/test_htmlparser.py @@ -38,17 +38,20 @@ class HTMLParserTreeBuilderSmokeTest(SoupTest, HTMLTreeBuilderSmokeTest): self.assertSoupEquals("foo &# bar", "foo &# bar") def test_tracking_line_numbers(self): - # Unlike other TreeBuilders, the html.parser TreeBuilder - # keeps track of line number and position of each element. - soup = self.soup( - "\n <p>\n\n<lineno>\n<b>text</b></lineno><offset></p>", - store_line_numbers=True - ) - self.assertEqual(2, soup.p.lineno) - self.assertEqual(3, soup.p.offset) - self.assertEqual("lineno", soup.p.find('lineno').name) - - + # The html.parser TreeBuilder keeps track of line number and + # position of each element. + markup = "\n <p>\n\n<sourceline>\n<b>text</b></sourceline><sourcepos></p>" + soup = self.soup(markup) + self.assertEqual(2, soup.p.sourceline) + self.assertEqual(3, soup.p.sourcepos) + self.assertEqual("sourceline", soup.p.find('sourceline').name) + + # You can deactivate this behavior. + soup = self.soup(markup, store_line_numbers=False) + self.assertEqual("sourceline", soup.p.sourceline.name) + self.assertEqual("sourcepos", soup.p.sourcepos.name) + + class TestHTMLParserSubclass(SoupTest): def test_error(self): """Verify that our HTMLParser subclass implements error() in a way |