def test_fix_html(self): # we can 'fix' HTML docs code = ("<html><head><title>t</title></head>" "<body><h1>head</h1></body></html>") proc = PSJHTMLProcessor() self.assertEqual( proc.fix_html(code), '<div id="psj-doc"><h1>head</h1></div>\n' )