1*4882a593Smuzhiyun"""Tests to ensure that the html.parser tree builder generates good 2*4882a593Smuzhiyuntrees.""" 3*4882a593Smuzhiyun 4*4882a593Smuzhiyunimport pickle 5*4882a593Smuzhiyunfrom bs4.testing import SoupTest, HTMLTreeBuilderSmokeTest 6*4882a593Smuzhiyunfrom bs4.builder import HTMLParserTreeBuilder 7*4882a593Smuzhiyun 8*4882a593Smuzhiyunclass HTMLParserTreeBuilderSmokeTest(SoupTest, HTMLTreeBuilderSmokeTest): 9*4882a593Smuzhiyun 10*4882a593Smuzhiyun @property 11*4882a593Smuzhiyun def default_builder(self): 12*4882a593Smuzhiyun return HTMLParserTreeBuilder() 13*4882a593Smuzhiyun 14*4882a593Smuzhiyun def test_namespaced_system_doctype(self): 15*4882a593Smuzhiyun # html.parser can't handle namespaced doctypes, so skip this one. 16*4882a593Smuzhiyun pass 17*4882a593Smuzhiyun 18*4882a593Smuzhiyun def test_namespaced_public_doctype(self): 19*4882a593Smuzhiyun # html.parser can't handle namespaced doctypes, so skip this one. 20*4882a593Smuzhiyun pass 21*4882a593Smuzhiyun 22*4882a593Smuzhiyun def test_builder_is_pickled(self): 23*4882a593Smuzhiyun """Unlike most tree builders, HTMLParserTreeBuilder and will 24*4882a593Smuzhiyun be restored after pickling. 25*4882a593Smuzhiyun """ 26*4882a593Smuzhiyun tree = self.soup("<a><b>foo</a>") 27*4882a593Smuzhiyun dumped = pickle.dumps(tree, 2) 28*4882a593Smuzhiyun loaded = pickle.loads(dumped) 29*4882a593Smuzhiyun self.assertTrue(isinstance(loaded.builder, type(tree.builder))) 30*4882a593Smuzhiyun 31*4882a593Smuzhiyun 32