Implemented line number tracking for html5lib.

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