From 5dbee3628972d521d44bdfb282ecc44fc956bdb1 Mon Sep 17 00:00:00 2001 From: Frazer Clews Date: Mon, 4 May 2020 14:26:18 +0100 Subject: lib/bs4/testing.py: fix bs4 testing Have classes actually inherit SoupTest so it can use it's original methods, also correctly import HTMLParserTreeBuilder Signed-off-by: Frazer Clews Signed-off-by: Richard Purdie --- lib/bs4/testing.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lib/bs4/testing.py b/lib/bs4/testing.py index 953bca8e9..6584ecf30 100644 --- a/lib/bs4/testing.py +++ b/lib/bs4/testing.py @@ -15,7 +15,7 @@ from bs4.element import ( SoupStrainer, ) -from bs4.builder import HTMLParserTreeBuilder +from bs4.builder._htmlparser import HTMLParserTreeBuilder default_builder = HTMLParserTreeBuilder @@ -56,7 +56,7 @@ class SoupTest(unittest.TestCase): self.assertEqual(earlier, e.previous_element) earlier = e -class HTMLTreeBuilderSmokeTest(object): +class HTMLTreeBuilderSmokeTest(SoupTest): """A basic test of a treebuilder's competence. @@ -541,7 +541,7 @@ Hello, world! data.a['foo'] = 'bar' self.assertEqual('text', data.a.decode()) -class XMLTreeBuilderSmokeTest(object): +class XMLTreeBuilderSmokeTest(SoupTest): def test_pickle_and_unpickle_identity(self): # Pickling a tree, then unpickling it, yields a tree identical -- cgit 1.2.3-korg